Spaces:
Sleeping
Sleeping
kaesb commited on
Commit ·
7ddb008
1
Parent(s): b75d5e6
Added description
Browse files
app.py
CHANGED
|
@@ -14,5 +14,8 @@ def predict(img):
|
|
| 14 |
img = PILImage.create(img)
|
| 15 |
pred,pred_idx,probs = learn.predict(img)
|
| 16 |
return {labels[i]: float(probs[i]) for i in range(len(labels))}
|
| 17 |
-
|
|
|
|
|
|
|
|
|
|
| 18 |
gr.Interface(fn=predict, inputs=gr.components.Image(shape=(512, 512)), outputs=gr.components.Label(num_top_classes=3)).launch()
|
|
|
|
| 14 |
img = PILImage.create(img)
|
| 15 |
pred,pred_idx,probs = learn.predict(img)
|
| 16 |
return {labels[i]: float(probs[i]) for i in range(len(labels))}
|
| 17 |
+
|
| 18 |
+
title = "Cat or dog?"
|
| 19 |
+
description = "Guesses whether the picture is of a cat or a dog"
|
| 20 |
+
|
| 21 |
gr.Interface(fn=predict, inputs=gr.components.Image(shape=(512, 512)), outputs=gr.components.Label(num_top_classes=3)).launch()
|