Spaces:
Runtime error
Runtime error
import gradio as gr | |
import os | |
from fastai.vision.all import load_learner | |
model = load_learner("model.pkl") | |
def predict(img): | |
prediction, idx, confidence = model.predict(img) | |
return prediction | |
image = gr.inputs.Image(shape=(200, 200), source="webcam") | |
label = gr.outputs.Label() | |
interface = gr.Interface(fn=predict, inputs=image, outputs=label) | |
interface.launch() |