update app.py

app.py

@@ -0,0 +1,27 @@
+from fastai.vision.all import load_learner
+import gradio as gr
+
+
+cap_labels=(
+    'baseball cap',
+    'beanie cap',
+    'cow boy hat',
+    'fedora cap',
+    'flat cap',
+    'kepi cap',
+    'newsboy cap',
+    'trucker cap'
+
+)
+model=load_learner(f'models/cap-classifier-v2.pkl')
+
+def recognize_image(image):
+  pred, idx, probs=model.predict(image)
+  print(pred)
+  return dict(zip(cap_labels, map(float, probs)))
+
+image = gr.Image(image_mode="RGB")
+label=gr.Label()
+example=['test_images/th (1).jpeg']
+iface = gr.Interface(fn=recognize_image, inputs=image, outputs=label,examples=example )
+iface.launch(inline=False, share=True)