Update app.py

app.py

@@ -1,7 +1,54 @@
+# import gradio as gr
+
+# def greet(name):
+#     return "Hello " + name + "!!"
+
+# iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+# iface.launch()
+
+from fastai.vision.all import *
 import gradio as gr
 
-def greet(name):
-    return "Hello " + name + "!!"
+# import pathlib
+# temp = pathlib.PosixPath
+# pathlib.PosixPath = pathlib.WindowsPath
+
+cap_labels = (
+    'balaclava cap', 
+     'baseball cap', 
+     'beanie cap', 
+     'boater hat', 
+     'bowler hat', 
+     'bucket hat', 
+     'cowboy hat', 
+     'fedora cap', 
+     'flat cap', 
+     'ivy cap', 
+     'kepi cap', 
+     'newsboy cap', 
+     'pork pie hat', 
+     'rasta cap', 
+     'sun hat', 
+     'taqiyah cap', 
+     'top hat', 
+     'trucker cap', 
+     'turban cap', 
+     'visor cap'
+)
+
+model = load_learner('cap-recognizer-v0.pkl')
+
+def recognize_image(image):
+    pred, idx, probs = model.predict(image)
+    return dict(zip(cap_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'
+    ]
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+iface = gr.Interface(recognize_image, gr.Image(shape=(192,192)), gr.Label(num_top_classes=5), examples=examples)
+iface.launch(inline=False)