Update app.py

app.py

@@ -49,8 +49,11 @@ def inference(img):
     pred_classes = prediction.topk(k=5).indices
     
     pred_class_names = [imagenet_id_to_classname[str(i.item())] for i in pred_classes[0]]
-    pred_class_probs = []
-    return "Top 5 predicted labels: %s" % ", ".join(pred_class_names)
+    pred_class_probs = [prediction[0][i.item()].item() * 100 for i in pred_classes[0]]
+    res = "Top 5 predicted labels:\n"
+    for name, prob in zip(pred_class_names, pred_class_probs):
+        res += f"{prob:.4f}\%:  {name}\n"
+    return res
     
 inputs = gr.inputs.Image(type='pil')
 outputs = gr.outputs.Textbox(label="Output")