initial commit

.gitattributes

@@ -29,3 +29,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
+export.pkl filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,33 @@
+# AUTOGENERATED! DO NOT EDIT! File to edit: ../bear_detector_using_gradio.ipynb.
+
+# %% auto 0
+__all__ = ['path', 'learn_inf', 'image', 'label', 'ui', 'classify_image']
+
+# %% ../bear_detector_using_gradio.ipynb 1
+import fastbook
+from fastbook import *
+import gradio as gr
+
+# %% ../bear_detector_using_gradio.ipynb 2
+# Load the model
+path = Path()
+learn_inf = load_learner(path/'export.pkl')
+
+# %% ../bear_detector_using_gradio.ipynb 3
+# Setup the UI
+image = gr.components.Image(shape=(192,192))
+label = gr.components.Label()
+
+def classify_image(image):
+    pred,pred_idx,probs = learn_inf.predict(image)
+    return f'Prediction: {pred}; Probability: {probs[pred_idx]:.04f}'
+
+ui = gr.Interface(
+    fn=classify_image,
+    inputs=image,
+    outputs=label,
+    examples=['horse.png', 'bear.png', 'cartoon.png'],
+    title='Bear Classifier',
+    description='A demo of exposing a model to the world with gradio and huggingfaces')
+
+ui.launch(inline=False)

export.pkl

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