import gradio as gr from fastai.vision.all import load_learner, PILImage import gradio as gr model = load_learner('model_res_94_7.pkl') def prediction(img): img_to_pred = PILImage.create(img) res,pos,prob = model.predict(img) return {res: format(prob[pos],'.4f')} gr.Interface(fn=prediction,inputs="image",outputs="label").launch()