Spaces:
Sleeping
Sleeping
| from fastai.vision.all import * | |
| import gradio as gr | |
| # import os | |
| # os.system("pip uninstall -y gradio") | |
| # os.system("pip install gradio==3.16.0") | |
| # import pathlib | |
| # temp = pathlib.PosixPath | |
| # pathlib.PosixPath = pathlib.WindowsPath | |
| model = load_learner('models/Recognizer.pkl') | |
| labels = ['Lion', 'Tiger'] | |
| def recognize_image(image): | |
| pred, id, probs = model.predict(image) | |
| return dict(zip(labels, map(float, probs))) | |
| image = gr.Image() | |
| output_labels = gr.Label() | |
| examples = [ | |
| 'examples/unknown_1.jpg', | |
| 'examples/unknown_2.jpg' | |
| ] | |
| iface = gr.Interface(fn=recognize_image, inputs = image, outputs = output_labels, examples = examples) | |
| # iface.queue(api_open=True) | |
| iface.launch(inline=False) | |