Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from fastai.vision.all import * | |
| import skimage | |
| import pickle | |
| learn = load_learner('snackdetector.pkl') | |
| labels = learn.dls.vocab | |
| def predict(img): | |
| img = PILImage.create(img) | |
| pred,pred_idx,probs = learn.predict(img) | |
| return {labels[i]: float(probs[i]) for i in range(len(labels))} | |
| title = "Dutch Snackbar Snack Detector" | |
| description = "Dutch Snackbar Snack Detector: Detect your Dutch snackbar snack and enjoy (or not)" | |
| article = "Snacks can be found everywhere in the Netherlands as they are often sold at stands as street food and as appetisers in bars and restaurants. This classifier is tied to savoury snacks which you most often find in Dutch snackbars. It ranges from the famous frikandel till the lesser -locally- known knoepert" | |
| examples = ['frikandel.jpg','kroket.jpg','bitterballen.jpg'] | |
| interpretation='default' | |
| enable_queue=True | |
| gr.Interface(fn=predict,inputs=gr.inputs.Image(shape=(224, 224)),outputs=gr.outputs.Label(num_top_classes=3),title=title,description=description,article=article,theme='huggingface',examples=examples,interpretation=interpretation,enable_queue=enable_queue).launch() |