init

.vscode/settings.json

@@ -0,0 +1,11 @@
+{
+    "files.exclude": {
+        "**/.git": true,
+        "**/.svn": true,
+        "**/.hg": true,
+        "**/CVS": true,
+        "**/.DS_Store": true,
+        "**/Thumbs.db": true
+    },
+    "hide-files.files": []
+}

app.py

@@ -0,0 +1,20 @@
+import gradio as gr
+from fastai.vision.all import *
+from pathlib import Path
+import pathlib
+hugging_face = True
+if not hugging_face:
+    temp = pathlib.PosixPath
+    pathlib.PosixPath = pathlib.WindowsPath
+learn = load_learner(Path("./model.pkl"))
+def predict_image(image):
+    pred, pred_idx, probs = learn.predict(image)
+    return pred, probs[pred_idx].item()
+
+app = gr.Interface(fn=predict_image, 
+    inputs=gr.Image(type="pil"), 
+    outputs=[gr.Label(), gr.Number()],
+    live=True
+)
+
+app.launch()

model.pkl

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

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+fastai
+timm