Spaces:
Runtime error
Runtime error
| from fastai.vision.all import * | |
| import gradio.components as gr | |
| import gradio | |
| import pathlib | |
| plt = platform.system() | |
| if plt == 'Windows': pathlib.PosixPath = pathlib.WindowsPath | |
| model = load_learner('./model.pkl') | |
| categories = ("Egyptian Mau", "Persian Cat", "Sphynx Cat") | |
| def classify(im): | |
| cat, idx, probs = model.predict(im) | |
| return dict(zip(categories, map(float, probs))) | |
| img = gr.Image() | |
| label = gr.Label() | |
| examples = ["persian.jpg", "egypian.jpg", "sphynix.jpg"] | |
| intr = gradio.Interface(fn=classify, inputs=img, outputs=label) | |
| intr.launch(inline=False) |