dancrvlh's picture
Adicionando o trabalho
8628d48
raw
history blame contribute delete
493 Bytes
from fastai.vision.all import *
import gradio as gr
learn = load_learner('Number_fastai.pkl')
categories = ('0','1','2','3','4','5','6','7','8','9')
def preds(img):
pred_class, _, probs = learn.predict(img)
return pred_class
image = gr.Image()
label = gr.Label()
examples = ['0.jpeg', '1.jpeg', '2.jpeg', '3.jpeg', '4.jpeg', '5.jpeg', '6.jpeg', '7.jpeg', '8.jpeg', '9.jpeg']
intf = gr.Interface(fn=preds, inputs=image, outputs=label, examples=examples)
intf.launch(inline=False)