Update files

README.md

@@ -4,11 +4,11 @@ emoji: 🐶
 colorFrom: pink
 colorTo: blue
 sdk: gradio
-sdk_version: 3.1.1
+sdk_version: 3.33.1
 app_file: app.py
 pinned: true
 license: apache-2.0
 duplicated_from: jph00/pets
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces#reference
+Kaggle Training notebook: https://www.kaggle.com/code/prasanth07/fastai-prd-apps-pets-training

app.py

@@ -1,27 +1,31 @@
-# AUTOGENERATED! DO NOT EDIT! File to edit: . (unless otherwise specified).
+# AUTOGENERATED! DO NOT EDIT! File to edit: app.ipynb.
 
-__all__ = ['learn', 'classify_image', 'categories', 'image', 'label', 'examples', 'intf']
+# %% auto 0
+__all__ = ['learn', 'categories', 'image', 'label', 'examples', 'intf', 'classify_image']
 
-# Cell
+# %% app.ipynb 2
 from fastai.vision.all import *
 import gradio as gr
 import timm
 
-# Cell
+# %% app.ipynb 4
 learn = load_learner('model.pkl')
 
-# Cell
+# %% app.ipynb 6
 categories = learn.dls.vocab
 
 def classify_image(img):
     pred,idx,probs = learn.predict(img)
     return dict(zip(categories, map(float,probs)))
 
-# Cell
+# %% app.ipynb 8
 image = gr.inputs.Image(shape=(192, 192))
 label = gr.outputs.Label()
 examples = ['basset.jpg']
 
-# Cell
-intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
-intf.launch()
+# %% app.ipynb 9
+intf = gr.Interface(
+    fn=classify_image, inputs=image, outputs=label, examples=examples,
+    title="Dog Breed Classifier",
+    description="Classifier is fine-tuned on pre-trained resnet34 model")
+intf.launch(inline=False)

model.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:02eb3a56c6194f7249d17c6ce92fc59c068a0c35e0fc3d552402eae8c7a0f346
-size 114778679
+oid sha256:4059bcf6ae28130f7b1f8aaaa1e2bf632ce598c1208cfa94c9e0e995c109e91d
+size 114793185

requirements.txt

@@ -1,3 +1,6 @@
-torch <1.12
-fastai>2.6.1
-timm
+fastai==2.7.12
+torch==2.0.1
+gradio==3.33.1
+timm==0.9.2
+numpy
+pandas