File size: 639 Bytes
a1d3bc3 c695ce5 a1d3bc3 c695ce5 a1d3bc3 c695ce5 a107f99 c695ce5 a1d3bc3 c695ce5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
from fastai.vision.all import *
import gradio as gr
learn = load_learner('model.pkl')
categories = ['calling', 'clapping', 'cycling', 'dancing', 'drinking', 'eating', 'fighting', 'hugging',
'laughing', 'listening_to_music', 'running', 'sitting', 'sleeping', 'texting', 'using_laptop']
def classify_image(img):
pred, idx, probs = learn.predict(img)
return {cat: float(prob) for cat, prob in zip(categories, probs)}
demo = gr.Interface(
fn=classify_image,
inputs=gr.Image(),
outputs=gr.Label(),
examples=[
'laughing.jpg',
'dancing.jpg',
'drinking.jpg'
]
)
demo.launch() |