bear_classifier_v1

app.py

@@ -0,0 +1,32 @@
+# AUTOGENERATED! DO NOT EDIT! File to edit: .
+
+# Cell
+__all__ = ['learn', 'categories', 'classify_image']
+
+
+# Cell
+from fastai.vision.widgets import *
+from fastai.vision.all import *
+import gradio as gr
+
+# Cell
+learn = load_learner("black_grizzly_teddy.pkl")
+
+# Cell
+categories = ('Black', 'Grizzly', 'Teddy')
+def classify_image(img):
+  pred, idx, probs = learn.predict(img)
+  return dict(zip(categories, map(float, probs)))
+
+# Cell
+image = gr.components.Image(shape= (192, 192))
+label = gr.components.Label()
+examples = ['black.jpg', 'grizzly.jpg', 'teddy.jpg']
+intf = gr.Interface(fn=classify_image,
+                    inputs=image,
+                    outputs=label,
+                    examples=examples)
+
+intf.launch(inline=False,
+            # share=True
+            )

black_grizzly_teddy.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b3b18d7856e84cd91762a42d81d5c9a29a5f7aac799e13a22cb0f96ba6b88ef9
+size 46970895