kaesb commited on
Commit
7ddb008
·
1 Parent(s): b75d5e6

Added description

Browse files
Files changed (1) hide show
  1. app.py +4 -1
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()