Update app.py

app.py

@@ -4,8 +4,8 @@ import gradio as gr
 learn = load_learner('model.pkl')
 
 def predict(img):
-    # fill in: call learn.predict(), build the dict for Gradio's Label output
-    pass
+    pred, pred_idx, probs = learn.predict(img)
+    return dict(zip(learn.dls.vocab, map(float, probs)))
 
 gr.Interface(
     fn=predict,