app.py

from fastai.vision.all import *
from fastai.vision.all import load_learner
import gradio as gr
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 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)