paddylab1 / app.py
SudheerNotes's picture
First Commit
ed34fe2
raw
history blame contribute delete
No virus
346 Bytes
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()