Update app.py

app.py

@@ -1,9 +1,26 @@
+from fastai.vision.all import *
+from fastai.vision.all import load_learner
 import gradio as gr
-import numpy as np
+from transformers import pipeline
+model=load_learner(f'models/cap-classifier-v2.pkl')
+img_path='test_images'
+cap_labels=(
+    'baseball cap',
+    'beanie cap',
+    'cow boy hat',
+    'fedora cap',
+    'flat cap',
+    'kepi cap',
+    'newsboy cap',
+    'trucker cap'
 
-
-def greet(name):
-    return "Hello " + name + "!!"
-
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch(share=True)
+)
+def recognize_image(image):
+  pred, idx, probs=model.predict(image)
+  print(pred)
+  return dict(zip(cap_labels, map(float, probs)))
+recognize_image(img)
+image = gr.Image(image_mode="RGB")
+label=gr.Label()
+iface = gr.Interface(fn=recognize_image, inputs=image, outputs=label,examples=example )
+iface.launch(inline=False, share=True)