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'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=['th (15).jpeg']
iface = gr.Interface(fn=recognize_image, inputs=image, outputs=label,examples=example )
iface.launch(inline=False, share=True)