boreavo commited on
Commit
47fda48
·
1 Parent(s): 1218c25
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -17,5 +17,9 @@ def predict(img):
17
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
18
 
19
 
20
- iface = gr.Interface(fn=predict, inputs="text", outputs="text")
21
- iface.launch()
 
 
 
 
 
17
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
18
 
19
 
20
+ iface = gr.Interface(
21
+ fn=predict,
22
+ inputs=gr.inputs.Image(shape=(512, 512)),
23
+ outputs=gr.outputs.Label(num_top_classes=3),
24
+ )
25
+ iface.launch(share=True)