Initial app

README.md

@@ -1,6 +1,6 @@
 ---
 title: Ugly Duckling Magic Mirror
-emoji: 📉
+emoji: 🪞🦢
 colorFrom: purple
 colorTo: blue
 sdk: gradio

app.py

@@ -0,0 +1,27 @@
+from fastai.vision.learner import load_learner
+import gradio as gr
+
+MODELPATH = Path().cwd().parent/'model'/'cygnet-vs-duckling.pkl'
+EXAMPLEPATH = Path().cwd().parent/'example-images'
+
+learn = load_learner(MODELPATH)
+categories = learn.dls.vocab
+
+def classify_image(image):
+    _, _, probs = learn.predict(image)
+    return dict(zip(categories, map(float, probs)))
+
+title = 'Mirror, mirror on the wall am I a duckling or a cygnet after all?'
+description = """Hans Christian Andersen's tale of the ugly duckling tells us about the sad youth of a cygnet acidentally being brought up in a family of ducks and being ostrized for its ugliness.\n But what if the poor cygnet had a magic mirror to tell it that it was in fact a young swan after all? Machine learning to the rescue!"""
+
+examples = EXAMPLEPATH.glob('*.jpg')
+
+app = gr.Interface(fn=classify_image,
+                   inputs=gr.components.Image(),
+                   outputs=gr.components.Label(),
+                   examples=examples,
+                   title=title,
+                   description=description,
+                   allow_flagging='never')
+
+app.launch()

model/cygnet-vs-duckling.pkl

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