Update app.py

app.py

@@ -104,7 +104,7 @@ def inference(input_image):
 
     # Show top categories per image
 
-    top1_prob, top1_catid = torch.topk(probabilities, 1)
+    top1_prob, top1_catid = torch.topk(probabilities, 2)
 
     result = {}
 
@@ -118,7 +118,7 @@ def inference(input_image):
 
 inputs = gr.inputs.Image(type='pil')
 
-outputs = gr.outputs.Label(type="confidences",num_top_classes=1)
+outputs = gr.outputs.Label(type="confidences",num_top_classes=2)