update app.py

app.py

@@ -1,7 +1,40 @@
 import gradio as gr
 
-def greet(name):
-    return "Hello " + name + "!!"
+cap_labels = cap_labels = {
+    'baseball cap',
+    'beanie cap',
+    'fedora cap',
+    'cowboy hat',
+    'kepi cap',
+    'flat cap',
+    'trucker cap',
+    # 'newsboy cap'
+    'pork pie hat',
+    'bowler hat',
+    'top hat',
+    'sun hat',
+    'boater hat',
+    # 'ivy cap',
+    'bucket hat',
+    'balaclava cap',
+    'turban cap',
+    'taqiyah cap',
+    'rasta cap',
+    'visor cap'
+}
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+def recognize_image(image):
+    pred, idx, probs = model.predict(image)
+    return dict(zip(sorted(cap_labels), map(float, probs)))
+
+image = gr.inputs.Image(shape=(192, 192))
+label = gr.outputs.Label()
+examples = [
+            'test_images/test_0.jpg',
+            'test_images/test_1.jpg',
+            'test_images/test_2.jpg', 
+            'test_images/test_3.jpg',
+            'test_images/test_4.jpg',
+            'test_images/test_5.jpg']
+iface = gr.Interface(fn=recognize_image, inputs=image, outputs=label, examples=examples)
+iface.launch(inline=False, share=True)