| import gradio as gr | |
| from fastai.vision.all import * | |
| from PIL import Image | |
| learn_inf = load_learner("export.pkl") | |
| def predict(value) -> str: | |
| img = Image.fromarray(value).convert('LA') | |
| pred,pred_idx,probs = learn_inf.predict(img) | |
| return f"{pred} at {probs[pred_idx]}" | |
| with gr.Blocks() as demo: | |
| with gr.Row(): | |
| with gr.Column(): | |
| input_img = gr.Image(label="Input", sources="webcam") | |
| with gr.Column(): | |
| output_lbl = gr.Label(value="Output", label="Expression Prediction") | |
| input_img.stream(fn=predict, inputs=input_img, outputs=output_lbl, time_limit=15, stream_every=0.1, concurrency_limit=30) | |
| if __name__ == "__main__": | |
| demo.launch() |