init

.gitattributes

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

.gitignore

@@ -0,0 +1,46 @@
+# Compiled Python files
+*.pyc
+*.pyo
+__pycache__/
+
+# Setuptools build output
+build/
+dist/
+
+# Python virtual environment
+venv/
+env/
+
+# IDE and editor files
+.vscode/
+.idea/
+*.pydevproject
+*.pyc
+*.bak
+*.swp
+*~.nib/
+*.sublime-project
+*.sublime-workspace
+*.komodoproject
+.komodotools/
+*_iPython/
+.ipynb_checkpoints/
+.sage/
+.spyderproject
+.spyproject
+.pytest_cache/
+
+# Django specific files
+*.log
+*.pot
+*.pyc
+local_settings.py
+
+# Database files
+*.sqlite3
+*.db
+
+# Other generated files
+*.bak
+*.swp
+*~

app.py

@@ -0,0 +1,20 @@
+from fastai.vision.all import *
+import gradio as gr
+
+
+loaded_model = load_learner('model.pkl')
+categories = loaded_model.dls.vocab
+
+
+def classify_image(img):
+ pred,idx,prob = loaded_model.predict(img)
+ return dict(zip(categories, map(float, prob)))
+
+
+image = gr.components.Image(type="pil")
+label = gr.components.Label()
+examples = ['multilabel_bird_on_horse.jpg']
+
+
+iface = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
+iface.launch(inline=False)

model.pkl

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

requirements.txt

@@ -0,0 +1,2 @@
+gradio
+fastai