DavidD003 commited on
Commit
9f90e61
·
1 Parent(s): ee88b66

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,6 +31,6 @@ def predict(img):
31
  otpt=F.softmax(learn.forward(inp_img.view(-1,64)))
32
  #pred,pred_idx,probs = learn.predict(img)
33
 
34
- return dict([[labels[i], float(otpt[0].data[i])] for i in range(len(labels))),inp_img
35
 
36
  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)
 
31
  otpt=F.softmax(learn.forward(inp_img.view(-1,64)))
32
  #pred,pred_idx,probs = learn.predict(img)
33
 
34
+ return dict([[labels[i], float(otpt[0].data[i])] for i in range(len(labels))]),inp_img
35
 
36
  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)