Feat: starting christmas vision space

README.md

@@ -1,6 +1,6 @@
 ---
 title: Christmas Panvision
-emoji: 🔥
+emoji: 🥂
 colorFrom: yellow
 colorTo: yellow
 sdk: gradio

app.py

@@ -0,0 +1,27 @@
+import gradio as gr
+# To solve problems with pathlib in load_learner
+import pathlib
+import platform
+plt = platform.system()
+if plt == 'Linux': pathlib.WindowsPath = pathlib.PosixPath
+
+from fastai.vision.all import load_learner, PILImage
+
+learn = load_learner('model.pkl')
+labels = learn.dls.vocab
+def predict(img):
+    img = PILImage.create(img)
+    pred,pred_idx,probs = learn.predict(img)
+    return {labels[i]: float(probs[i]) for i in range(len(labels))}
+
+title = "Panettone or Pandoro?"
+description = "A prototype classifier developed with fastaiwith images from ddg."
+examples = [r'examples/panettone_01.jpg',
+            r'examples/panettone_02.jpg'
+            r'examples/pandoro_01.jpg'
+            r'examples/pandoro_02.jpg'
+            ]
+
+
+demo = gr.Interface(fn=predict, inputs="image", outputs="label", examples=examples)
+demo.launch()

model.pkl

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

requirements.txt

@@ -0,0 +1 @@
+fastai==2.7.13