Update app.py

app.py

@@ -12,7 +12,7 @@ labels = learn.dls.vocab
 
 def predict(img):
     pred, _, probs = learn.predict(img)
-    return f"This is a {pred} with probability {probs[pred_idx]}"
+    return pred
 
 # Create the Gradio interface
 iface = gr.Interface(