Update app.py

app.py

@@ -47,7 +47,7 @@ def inference(input_image):
     with open("dog_cat.txt", "r") as f:
         categories = [s.strip() for s in f.readlines()]
     # Show top categories per image
-    top5_prob, top5_catid = torch.topk(probabilities, 1)
+    top5_prob, top5_catid = torch.topk(probabilities, 2)
     result = {}
     for i in range(top5_prob.size(0)):
         result[categories[top5_catid[i]]] = top5_prob[i].item()