File size: 434 Bytes
063df12
27e2a2c
bdfca3e
 
27e2a2c
 
063df12
bdfca3e
 
 
 
 
 
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():
    return "Hello, World!"

iface = gr.Interface(fn=greet, inputs=[], outputs="text")
iface.launch()