Spaces:
Sleeping
Sleeping
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() |