app.py

# AUTOGENERATED! DO NOT EDIT! File to edit: loaded_model.ipynb.

# %% auto 0
__all__ = ['model', 'labels', 'image', 'label', 'examples', 'em_model', 'emotion_recognition']

# %% loaded_model.ipynb 1
import gradio as gr
from fastai.vision.all import *

# %% loaded_model.ipynb 4
model=load_learner('em_model.pkl')

# %% loaded_model.ipynb 5
labels=model.dls.vocab

# %% loaded_model.ipynb 6
labels=tuple(labels)

# %% loaded_model.ipynb 14
def emotion_recognition(img):
    emotion,_,prob=model.predict(img)
    return  {labels[i]: float(prob[i]) for i in range(len(labels))}

# %% loaded_model.ipynb 17
image=gr.Image(height=192,width=192)
label=gr.Label()
examples=['sample_face.jpg','sample_face2.jpg']

em_model=gr.Interface(fn=emotion_recognition,inputs=image,outputs=label,examples=examples)
em_model.launch(inline=False)