model and images added

README.md

@@ -4,10 +4,14 @@ emoji: 📈
 colorFrom: pink
 colorTo: pink
 sdk: gradio
-sdk_version: 4.31.0
+sdk_version: 3.50.0
 app_file: app.py
 pinned: false
 license: mit
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+
+HuggingFace Spaces App URL: https://huggingface.co/spaces/msideadman/cap-recognizer
+
+Gradio App URL: https://f6f295e355f73e92a8.gradio.live

app.py

@@ -1,7 +1,53 @@
+from fastai.vision.all import *
 import gradio as gr
 
-def greet(name):
-    return "Hello " + name + "!!"
+pasta_shape_labels = (
+    "bucatini",
+    "cannelloni",
+    "cavatappi",
+    "conchiglie",
+    "farfalle",
+    "fettuccine",
+    "fusilli",
+    "gemelli",
+    "lasagna",
+    "linguine",
+    "macaroni",
+    "orecchiette",
+    "orzo",
+    "penne",
+    "ravioli",
+    "rigatoni",
+    "rotini",
+    "spaghetti",
+    "tagliatelle",
+    "tortellini"
+)
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+model = load_learner('pasta_shape_recognizer_v2.pkl')
+
+def recognize_image(image):
+    pred, idx, probs = model.predict(image)
+    return dict(zip(pasta_shape_labels, map(float, probs)))
+
+image = gr.inputs.Image(shape=(192,192))
+label = gr.outputs.Label(num_top_classes=5)
+examples = [
+    'unknown00.png',
+    'unknown01.png',
+    'unknown02.png',
+    'unknown03.png',
+    'unknown04.png',
+    'unknown05.png',
+    'unknown06.png',
+    'unknown07.png',
+    'unknown08.png',
+    'unknown09.png',
+    'unknown10.png',
+    'unknown11.png',
+    'unknown12.png',
+    'unknown13.png'
+]
+
+iface = gr.Interface(fn=recognize_image, inputs=image, outputs=label, examples=examples)
+iface.launch(inline=False)

pasta_shape_recognizer_v2.pkl

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

requirements.txt

@@ -0,0 +1,2 @@
+fastai==2.7.15
+gradio==3.50.0