BirdOrForest / app.py
3v324v23's picture
Commit : 103
bf56825
raw
history blame
402 Bytes
import gradio as gr
from fastbook import *
learn = load_learner('BirdOrForest.pkl')
image = gr.inputs.Image(shape=(224, 224))
label = gr.outputs.Label()
path = Path("Examples")
examples = [[path/"1.jpg"], [path/"2.jpg"], [path/"3.jpg"], [path/"4.jpg"]]
def prediction(img):
return
demo = gr.Interface(fn=prediction, inputs=image, outputs=label, examples=examples)
demo.launch(inline = False)