File size: 579 Bytes
b198f48 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
from fastai.vision.all import *
import gradio as gr
learn = load_learner("hotdogModel.pkl")
def classify_image(image: Image):
prediction, index, probability = learn.predict(image)
return "is_hotdog.png" if prediction == "hotdog" else "not_hotdog.png"
examples = ["examples/hotdog.jpg", "examples/burger.jpg", "examples/sandwich.jpg", "examples/dog.jpg",
"examples/hotdog_dog.jpg", "examples/fancy_hotdog.jpg"]
iface = gr.Interface(fn=classify_image, inputs="image", outputs="image", examples=examples, allow_flagging="never")
iface.launch(inline=False)
|