Spaces:
Runtime error
Runtime error
| import gradio as gr | |
| from fastai.vision.all import load_learner | |
| model = load_learner('export.pkl') | |
| categories = ('george', 'not_george') | |
| def classify_bird(img): | |
| pred, idx, probs = model.predict(img) | |
| return dict(zip(categories, map(float, probs))) | |
| image = gr.inputs.Image(shape=(256, 256)) | |
| label = gr.outputs.Label(num_top_classes=5) | |
| examples = ['george1.jpeg', 'george2.jpg', 'george3.jpeg', 'not_george1.jpg', 'not_george2.jpg', 'not_george3.jpg'] | |
| iface = gr.Interface( | |
| fn=classify_bird, | |
| inputs=image, | |
| outputs=label, | |
| examples=examples, | |
| title='George Classifier', | |
| description='Determines whether an image is of george', | |
| allow_flagging='never' | |
| ) | |
| iface.launch() |