File size: 344 Bytes
b5e7e15 20d1cb7 b5e7e15 20d1cb7 b5e7e15 20d1cb7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
import gradio as gr
from fastai.vision.all import *
learn = load_learner('model.pkl')
categories = ('Cockapoo', "Labradoodle")
def predict(img):
pred, idx, probs = learn.predict(img)
return dict(zip(categories, map(float, probs)))
iface = gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(192,192)), outputs="text")
iface.launch() |