initial version

.gitignore

@@ -0,0 +1,3 @@
+flagged/
+.ipynb_checkpoints/
+.vscode/

README.md

@@ -1,13 +1,15 @@
 ---
 title: Fastai Classifier
 emoji: 😻
-colorFrom: green
-colorTo: pink
+colorFrom: blue
+colorTo: green
 sdk: gradio
 sdk_version: 3.9
 app_file: app.py
-pinned: false
+pinned: true
 license: apache-2.0
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# Dragon or monster
+
+This is a simple exercise to detect "monster" or "dragon" trained on ResNet.

app.py

@@ -0,0 +1,24 @@
+from fastai.vision.all import *
+import gradio as gr
+
+
+# Cell
+learn = load_learner("model-resnet18.pkl")
+
+# Cell
+categories = learn.dls.vocab
+
+
+def classify_image(img):
+    pred, idx, probs = learn.predict(img)
+    return dict(zip(categories, map(float, probs)))
+
+
+# Cell
+image = gr.inputs.Image(shape=(256, 256))
+label = gr.outputs.Label()
+
+examples = ["dragon.jpg", "monster.jpg"]
+
+app = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
+app.launch()

model-resnet18.pkl

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

model-resnet50.pkl

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

requirements.txt

@@ -0,0 +1,2 @@
+torch <1.12
+fastai>2.6.1