using model

app.py

@@ -1,7 +1,16 @@
 import gradio as gr
 
-def greet(name):
-    return "Hello " + name + "!!"
+import fastbook
+fastbook.setup_book()
+from fastbook import *
+from fastai.vision.widgets import *
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+f = Path() / 'bamboo_or_butterfly.pkl'
+learn_inf = load_learner(f)
+
+def predict(img)
+    pred, pred_idx, probs = learn_inf.predict(img.to_thumb(224))
+    f'Prediction: {pred}; Probability:{probs[pred_idx]:.04f}'
+
+iface = gr.Interface(fn=predict, inputs="image", outputs="text")
 iface.launch()