bear_classif_portal

app.py

@@ -1,7 +1,14 @@
+from fastai.vision.all import *
 import gradio as gr
 
-def greet(name):
-    return f"Hello {name}!"
+learn = load_learner("model.pkl")
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+categories = {'Grizzly bear','black bear','teddy bear'}
+def classify_image(img):
+  pred,idx,probs = learn.predict(img)
+  return dict(zip(categories, map(float,probs)))
+
+image = gr.inputs.Image(shape=(192, 192))
+label = gr.outputs.Label()
+Examples = ['teddybear.jpg','blackbear.jpg','grizzlybear.jpg']
+gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=Examples).launch()