| #!/usr/bin/env python | |
| # coding: utf-8 | |
| # In[1]: | |
| import gradio as gr | |
| from fastai.vision.all import load_learner | |
| # In[33]: | |
| # In[4]: | |
| learner = load_learner("export.pkl") | |
| # In[46]: | |
| categories = ["black", "grizzly", "teddy"] | |
| def image_classifier(img): | |
| pred, idx, probs = learner.predict(img) | |
| return dict(zip(categories, map(float, probs))) | |
| # In[47]: | |
| # In[48]: | |
| intf = gr.Interface( | |
| fn=image_classifier, | |
| inputs="image", | |
| outputs="label", | |
| examples=["black.jpg", "grizzly.jpg", "teddy.jpg"], | |
| ) | |
| intf.launch() | |
| # ## Export | |
| # In[ ]: | |