Update app.py

app.py

@@ -14,16 +14,18 @@ def get_x(r):
     return path/r["fname"]
     
 learner = from_pretrained_fastai(repo_id)
+labels = learner.dls.vocab
 
 def inference(image):
     label_predict,_,probs = learner.predict(image)
-    return f"This grapevine leave is {label_predict} with {100*probs[torch.argmax(probs)].item():.2f}% probability"
+    labels_probs = {labels[i]: float(probs[i]) for i, _ in enumerate(labels)}
+    return labels_probs
 
 gr.Interface(
     fn=inference,
     title="Grapevine leave image classification",
     description = "Predict which type of grapevine leave belong to Ak, Ala_Idris, Buzgulu, Dimnit, Nazli",
-    inputs="image",
+    inputs=output = gr.outputs.Label(num_top_classes=3),
     examples=examples,
     outputs=gr.Textbox(label='Prediction'),
     cache_examples=False,