Initial model deploy

.gitattributes

@@ -32,3 +32,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.png filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,29 @@
+__all__ = ["examples", "iface", "learn", "labels", "classify_bear", "bear_image", "outputs"]
+
+import gradio as gr
+from fastai.vision.all import *
+from fastcore.all import *
+
+learn = load_learner("model/export.pkl")
+labels = learn.dls.vocab
+
+examples = [
+    "examples/black.jpg",
+    "examples/grizzly.jpg",
+    "examples/panda.jpg",
+    "examples/polar.jpg",
+    "examples/teddy.png"
+]
+
+def classify_bear(img):
+    img = PILImage.create(img)
+    pred,idx,probs = learn.predict(img)
+    return f"Prediction: {pred}; Probability: {probs[idx]:.04f}"
+
+bear_image = gr.inputs.Image(shape=(192,192))
+outputs = gr.outputs.Label(num_top_classes=5)
+
+# App launch
+iface = gr.Interface(
+    fn=classify_bear, inputs=bear_image, outputs=outputs, examples=examples)
+iface.launch()

model/export.pkl

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

requirements.txt

@@ -0,0 +1,3 @@
+fastai
+scikit-image
+gradio