Update app.py

app.py

@@ -22,7 +22,8 @@ learner = from_pretrained_fastai(repo_id)
 
 def inference(image):
     label_predict,_,probs = learner.predict(image)
-    return f"This rice image 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,
@@ -30,7 +31,7 @@ gr.Interface(
     description = "Predict which type of rice belong to Arborio, Basmati, Ipsala, Jasmine, Karacadag",
     inputs="image",
     examples=examples,
-    outputs=gr.Textbox(label='Prediction'),
+    output=gr.outputs.Label(num_top_classes=5, label='Prediction'),
     cache_examples=False,
     article = "Author: <a href=\"https://www.linkedin.com/in/vumichien/\">Vu Minh Chien</a>",
 ).launch(debug=True, enable_queue=True)