DavidD003 commited on
Commit
831cba9
·
1 Parent(s): 1cc47bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -61,4 +61,4 @@ def predict(img):
61
 
62
  return dict([[labels[i], float(otpt[0].data[i])] for i in range(len(labels))]),inp_img
63
 
64
- gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(512, 512)), outputs=[gr.outputs.Label(num_top_classes=10), gr.outputs.Image()]).launch(share=True)
 
61
 
62
  return dict([[labels[i], float(otpt[0].data[i])] for i in range(len(labels))]),inp_img
63
 
64
+ gr.Interface(fn=predict, inputs=gr.inputs.Image(type='filepath'), outputs=[gr.outputs.Label(num_top_classes=10), gr.outputs.Image()]).launch(share=True)