DavidD003 commited on
Commit
f551867
·
1 Parent(s): fccc9a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ labels = learn.dls.vocab
7
  def predict(img):
8
  img = PILImage.create(img)
9
  pred,pred_idx,probs = learn.predict(img)
10
- return "Result"#{labels[i]: float(probs[i]) for i in range(len(labels))}
11
 
12
  gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(512, 512)), outputs=gr.outputs.Label(num_top_classes=3)).launch(share=True)
13
  #gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(512, 512)), outputs=gr.outputs.Label()).launch(share=True)
 
7
  def predict(img):
8
  img = PILImage.create(img)
9
  pred,pred_idx,probs = learn.predict(img)
10
+ return {labels[i]: float(probs[i]) for i in range(len(labels))}
11
 
12
  gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(512, 512)), outputs=gr.outputs.Label(num_top_classes=3)).launch(share=True)
13
  #gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(512, 512)), outputs=gr.outputs.Label()).launch(share=True)