Update app.py

app.py

@@ -26,7 +26,7 @@ def predict(img):
     probability = torch.nn.functional.softmax(out[0],dim=0)
 
     values, indices = torch.topk(probability,k=5)
-    return {LABELS[i]: v.items() for i,v in zip(indices,values)}
+    return {LABELS[i]: v.item() for i,v in zip(indices,values)}
 
 
 iface = gr.Interface(fn=predict, inputs=gr.inputs.Image(type='pil'), outputs="label").launch()