pets

app.py

@@ -1,9 +1,8 @@
-__all__ = ['learn', 'classify_image', 'categories', 'image', 'label', 'examples', 'intf']
-
 # Cell
+import timm
 from fastai.vision.all import *
 import gradio as gr
-import timm
+
 
 # Cell
 learn = load_learner('model.pkl')
@@ -16,10 +15,10 @@ def classify_image(img):
     return dict(zip(categories, map(float,probs)))
 
 # Cell
-image = gr.inputs.Image(shape=(192, 192))
-label = gr.outputs.Label()
+# 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 = gr.Interface(fn=classify_image, inputs="image", outputs="label", examples=examples)
 intf.launch()

model.pkl

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

requirements.txt

@@ -0,0 +1,6 @@
+fastai
+gradio==3.34.0
+timm==0.6.13
+torch==2.0.1
+torchvision==0.15.2
+torch==1.12.1