Upload folder using huggingface_hub

README.md

@@ -1,12 +1,6 @@
 ---
-title: Fastai Season Guesser
-emoji: 💻
-colorFrom: yellow
-colorTo: pink
-sdk: gradio
-sdk_version: 4.13.0
+title: fastai-season-guesser
 app_file: app.py
-pinned: false
+sdk: gradio
+sdk_version: 4.11.0
 ---
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,29 @@
+from fastai.vision.all import load_learner, PILImage
+from pathlib import Path
+import gradio
+
+here = Path(__file__).parent
+predicter = load_learner(here / 'seasons.pkl')
+
+labels = predicter.dls.vocab
+
+def predict(img):
+    _img = PILImage.create(img)
+    _, _, probabilities = predicter.predict(_img)
+    return {labels[i]: float(prob) for i, prob in enumerate(probabilities)}
+
+example_dir = here / 'test'
+gradio.Interface(
+    fn=predict,
+    inputs=gradio.Image(),
+    outputs=gradio.Label(num_top_classes=3),
+    title='Season Guesser',
+    description='What season is this image?',
+    article="<p style='text-align: center'><a href='https://github.com/exvertus/fastai-coursework/tree/main/one/seasons' target='_blank'>See the code</a></p>",
+    examples=[
+        example_dir / 'summer.jpg',
+        example_dir / 'winter.jpg',
+        example_dir / 'autumn.jpg',
+        example_dir / 'winter-art.jpg',
+    ]
+).launch()

requirements.txt

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

seasons.pkl

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