Update app.py

app.py

@@ -4,7 +4,7 @@ import gradio as gr
 learn_inf = load_learner('export.pkl')
 def classify_image(img):
     pred,pred_idx,probs = learn_inf.predict(img)
-    return f'Prediction: {pred}; Probability: {probs[pred_idx]:.04f}'
+    return dict(zip(pred, map(float, probs[pred_idx])))
     
 image = gr.inputs.Image(shape=(224, 224))
 label = gr.outputs.Label()