Spaces:
Runtime error
Runtime error
File size: 427 Bytes
f33e051 8f0c44b c16e60b f33e051 c16e60b f33e051 ce32175 f33e051 ce32175 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
import gradio as gr
from fastai.vision.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
interface = gr.Interface(fn=predict,
inputs=gr.inputs.Image(shape=(512, 512)),
outputs=gr.outputs.Label(),
interpretation="default")
interface.launch() |