Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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)
|