Spaces:
Runtime error
Runtime error
File size: 386 Bytes
f33e051 809f2cc c16e60b f33e051 c16e60b f33e051 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
import gradio as gr
from fastcore.all import *
learner_inf = load_learner('baltrishkes.pkl')
def predict(img):
img = PILImage.create(img)
pred,pred_idx,probs = learner_inf.predict(img)
return pred
gr.Interface(fn=predict,
inputs=gr.inputs.Image(shape=(512, 512)),
outputs=gr.outputs.Label(),
interpretation="default").launch() |