Changed to model pth since huggingface does not allow pkl anymore

.gitattributes

@@ -34,3 +34,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
 model.pkl filter=lfs diff=lfs merge=lfs -text
+model.pth filter=lfs diff=lfs merge=lfs -text

app.py

@@ -8,12 +8,20 @@
 #demo = gr.Interface(fn=greet, inputs="text", outputs="text")
 #demo.launch()
 
+# second time used this only but huggingface doesn't like it so moving to a different method
+#learner=load_learner('model.pkl')
+
 from fastai.vision.all import *
 import gradio as gr
 
 def is_cat(x): return x[0].isupper()
 
-learner=load_learner('model.pkl')
+dls = ImageDataLoaders.from_lists('.', fnames=['cat.jpeg','dog.jpeg'], labels=['cat','dog'], vocab=['cat', 'dog'])
+
+learner = vision_learner(dls, resnet18, metrics=error_rate)
+
+learner.load('model')
+
 
 categories = ['Dog','Cat']
 def classify_image(img):

models/model.pth

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