Update app.py

app.py

@@ -62,7 +62,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]]] = top5_prob[i].item()
+        result[categories[top5_catid[i].item()]] = top5_prob[i].item()
     return result
 
 inputs = gr.Image(type='pil')