app with pepper-classifier model

app.py

@@ -1,8 +1,26 @@
+# AUTOGENERATED! DO NOT EDIT! File to edit: pepper-classifier-app.ipynb.
+
+# %% auto 0
+__all__ = ['learner', 'pepper_types', 'image', 'label', 'examples', 'interface', 'predict']
+
+# %% pepper-classifier-app.ipynb 0
+from fastai.vision.all import *
 import gradio as gr
 
-def greet(name):
-    return "Hello " + name + "!!"
+# %% pepper-classifier-app.ipynb 1
+learner = load_learner('pepper-classifier-model.pkl')
+
+# %% pepper-classifier-app.ipynb 2
+pepper_types = 'italian', 'morron', 'padron'
+
+def predict(image):
+    type_of_pepper,_,probs = learner.predict(image)
+    return dict(zip(pepper_types, map(float, probs)))
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+# %% pepper-classifier-app.ipynb 7
+image = gr.inputs.Image(shape=(224, 224))
+label = gr.outputs.Label()
+examples = ['pimiento-morron-rojo.jpg', 'pimiento-italiano-rojo.jpg', 'pimientos-del-padron.jpg', 'pimiento-italiano-verde.jpg']
 
+interface = gr.Interface(fn=predict, inputs=image, outputs=label, examples=examples)
+interface.launch(inline=False)

pepper-classifier-model.pkl

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