Deploy pet breed classifier

.gitignore

@@ -0,0 +1,146 @@
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+env/
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+*.egg-info/
+.installed.cfg
+*.egg
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*,cover
+.hypothesis/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+target/
+
+# IPython Notebook
+.ipynb_checkpoints
+
+# pyenv
+.python-version
+
+# celery beat schedule file
+celerybeat-schedule
+
+# dotenv
+.env
+
+# virtualenv
+venv/
+ENV/
+
+# Spyder project settings
+.spyderproject
+
+# Rope project settings
+.ropeproject
+### VirtualEnv template
+# Virtualenv
+# http://iamzed.com/2009/05/07/a-primer-on-virtualenv/
+.Python
+[Bb]in
+[Ii]nclude
+[Ll]ib
+[Ll]ib64
+[Ll]ocal
+[Ss]cripts
+pyvenv.cfg
+.venv
+pip-selfcheck.json
+### JetBrains template
+# Covers JetBrains IDEs: IntelliJ, RubyMine, PhpStorm, AppCode, PyCharm, CLion, Android Studio and Webstorm
+# Reference: https://intellij-support.jetbrains.com/hc/en-us/articles/206544839
+
+# User-specific stuff:
+.idea/workspace.xml
+.idea/tasks.xml
+.idea/dictionaries
+.idea/vcs.xml
+.idea/jsLibraryMappings.xml
+
+# Sensitive or high-churn files:
+.idea/dataSources.ids
+.idea/dataSources.xml
+.idea/dataSources.local.xml
+.idea/sqlDataSources.xml
+.idea/dynamic.xml
+.idea/uiDesigner.xml
+
+# Gradle:
+.idea/gradle.xml
+.idea/libraries
+
+# Mongo Explorer plugin:
+.idea/mongoSettings.xml
+
+.idea/
+
+## File-based project format:
+*.iws
+
+## Plugin-specific files:
+
+# IntelliJ
+/out/
+
+# mpeltonen/sbt-idea plugin
+.idea_modules/
+
+# JIRA plugin
+atlassian-ide-plugin.xml
+
+**/.DS_Store
+
+.vscode/

app.py

@@ -0,0 +1,37 @@
+import gradio as gr
+from fastai.vision.all import *
+import skimage
+
+learn = load_learner("model/model.pkl")
+
+labels = learn.dls.vocab
+
+
+def predict(img):
+    img = PILImage.create(img)
+    pred, pred_idx, probs = learn.predict(img)
+    return dict(zip(labels, map(float, probs)))
+
+
+title = "Pet Breed Classifier"
+description = "A pet breed classifier trained on the Oxford Pets dataset with fastai."
+article = "<p style='text-align: center'><a href='https://github.com/mchen50' target='_blank'>My Github</a></p>"
+examples = [
+    "examples/siamese.jpg",
+    "examples/leonberger.jpg",
+    "examples/leonberger.jpg",
+]
+interpretation = "default"
+
+app = gr.Interface(
+    fn=predict,
+    inputs=gr.Image(shape=(512, 512)),
+    outputs=gr.Label(num_top_classes=3),
+    title=title,
+    description=description,
+    article=article,
+    examples=examples,
+    interpretation=interpretation,
+)
+app.queue()
+app.launch()

makefile

@@ -0,0 +1,31 @@
+SHELL := /bin/bash
+
+.ONESHELL:
+
+setup: clean install
+
+install:
+	python -m venv venv; \
+	source venv/bin/activate; \
+	pip install -r requirements.txt \
+
+test: clean
+	source venv/bin/activate; \
+	pytest tests/
+
+clean-all: clean
+	rm -rf venv; 
+
+clean:
+	rm -rf .coverage; \
+	rm -rf .pytest_cache; \
+	rm -rf tests/.pytest_cache; \
+	find . -type f -name '*.py[co]' -delete -o -type d -name __pycache__ -delete
+
+check-format:
+	source venv/bin/activate; \
+	black --check --diff .
+
+format:
+	source venv/bin/activate; \
+	black .

model/model.pkl

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

requirements.txt

@@ -0,0 +1,2 @@
+fastai
+scikit-image