Spaces:
Sleeping
Sleeping
import platform | |
import pathlib | |
if platform.system() == "Windows": | |
temp = pathlib.PosixPath | |
pathlib.PosixPath = pathlib.WindowsPath | |
from fastai.vision.all import * | |
import gradio as gr | |
learn = load_learner(Path("model.pkl")) | |
def classify_image(img): | |
pred, idx, probs = learn.predict(img) | |
return f"This is {pred}.\nConfidence Level: {float(max(probs))}" | |
image = gr.Image(shape=(192, 192)) | |
examples = ['images/pizza.jpg', 'images/pasta.jpg', 'images/dunno.jpg'] | |
app = gr.Interface(fn=classify_image, | |
inputs=image, | |
outputs=gr.Textbox(label="Output"), | |
examples=examples) | |
app.launch() | |