| import gradio as gr | |
| from fastai.vision.all import * | |
| import skimage | |
| learn = load_learner('characters.pkl') | |
| labels = learn.dls.vocab | |
| def predict(img): | |
| pred, pred_idx, probs = learn.predict(img) | |
| return {labels[i]: float(probs[i]) for i in range(len(labels))} | |
| title = "Video Game Character Classifier" | |
| examples = ['ellie.jpg', 'arthur.jpg', 'kratos.jpg', 'ellielou.jpg'] | |
| # Updated Gradio Interface with new syntax | |
| with gr.Blocks() as demo: | |
| gr.Markdown("# " + title) | |
| image_input = gr.Image(shape=(128, 128)) | |
| label_output = gr.Label() | |
| # Create submit button | |
| submit_btn = gr.Button("Classify") | |
| submit_btn.click(fn=predict, inputs=image_input, outputs=label_output) | |
| # Adding examples | |
| gr.Examples(examples, inputs=image_input, outputs=label_output, fn=predict) | |
| demo.launch() |