bear classification

app.py

@@ -1,7 +1,12 @@
 import gradio as gr
+from huggingface_hub import from_pretrained_fastai
 
-def greet(name):
-    return "Hello " + name + "!!"
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+learner = from_pretrained_fastai("hdchild/bear")
+
+def predict(image):
+    return learner.predict(image)[0]
+
+
+iface = gr.Interface(fn=predict, inputs="image", outputs="text", capture_session=True)
 iface.launch()