| import gradio as gr | |
| from fastai.vision.all import * | |
| from PIL import Image | |
| learn_inf = load_learner("export.pkl") | |
| def predict(value: Image.Image) -> str: | |
| pred,pred_idx,probs = learn_inf.predict(value) | |
| 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() |