Adding files

README.md

@@ -1,12 +1,13 @@
 ---
-title: Pets Classifier
-emoji: 📊
-colorFrom: green
-colorTo: blue
+title: Fastai Pet Classifier
+emoji: 🏢
+colorFrom: pink
+colorTo: purple
 sdk: gradio
-sdk_version: 4.38.1
+sdk_version: 4.29.0
 app_file: app.py
 pinned: false
+license: apache-2.0
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,21 @@
+import gradio as gr
+from fastai.vision.all import *
+import skimage
+
+#learn =  load_learner('model.pkl')
+learn =  load_learner('model_convnext.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 = "Pet Breed Classifier"
+description = "A pet breed classifier trained on the Oxford Pets dataset with fastai. Created as a demo for Gradio and HuggingFace Spaces."
+examples = ['shiba.jpeg','ragdoll.jpeg']
+#interpretation='default'
+#enable_queue=True
+
+gr.Interface(fn=predict,inputs=gr.Image(type="pil"),outputs=gr.Label(num_top_classes=3)
+             ,title=title,description=description,examples=examples).launch()

model.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cee4c6bc7726119e93d954045f4f82caaf60fc7bdf7bcbdc63e6c075af5fc75d
+size 103096018

model_convnext.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3c77df2d1288ae5bcfd1292586d7cf734806a86a02b49418d17c017bc7505a26
+size 354985524

requirements.txt

@@ -0,0 +1,3 @@
+fastai
+scikit-image
+timm