Spaces:
Sleeping
Sleeping
#!/usr/bin/env python | |
# coding: utf-8 | |
# In[1]: | |
import gradio as gr | |
from fastai.vision.all import * | |
# In[3]: | |
def breed_name(x): | |
return ''.join([char for char in x if not char.isdigit()][:-5]) | |
learn = load_learner('pet_breeds.pkl') | |
# In[4]: | |
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))} | |
# In[5]: | |
gr.Interface(fn=predict, | |
inputs=gr.components.Image(height=512, width=512), | |
outputs=gr.components.Label(num_top_classes=3), | |
title='What breed is it ?', | |
description='A pet breeds classifier', | |
article="<p style='text-align: center'><a href='https://www.tanishq.ai/blog/posts/2021-11-16-gradio-huggingface.html' target='_blank'>Reference</a></p>", | |
examples=['keeshond.jpeg', 'maine_coon.jpeg'] | |
).launch(share=True) | |
# In[ ]: | |