Update app.py

app.py

@@ -61,7 +61,7 @@ def inference(input_image):
     top5_prob, top5_catid = torch.topk(probabilities, 6)
     result = {}
     for i in range(top5_prob.size(0)):
-        result[categories[top5_catid[i],item()]] = top5_prob[i].item()
+        result[categories[top5_catid[i].item()]] = top5_prob[i].item()
     return result
 
 inputs = gr.Image(type='pil')