Spaces:
Runtime error
Runtime error
File size: 459 Bytes
063df12 27e2a2c bdfca3e 27e2a2c 063df12 bdfca3e 4aa1057 063df12 4aa1057 063df12 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
import gradio as gr
from fastai.learner import load_learner
from fastai.vision.core import PILImage
from PIL import Image
learner = load_learner('export.pkl')
def predict_bear_type(img_path):
img = Image.open(img_path)
img = PILImage.create(img)
pred,pred_idx,probs = learner.predict(img)
return pred
def greet(img_path):
return predict_bear_type(img_path)
iface = gr.Interface(fn=greet, inputs="image", outputs="text")
iface.launch()
|