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