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()