Spaces:

mpiendl
/

pet_classifier

Runtime error

mpiendl commited on Dec 12, 2023

Commit

f60e1f4

1 Parent(s): e623f7f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,8 +1,9 @@
 import gradio as gr
 from fastai.vision.all import *
 import skimage
-learn = load_learner('export.pkl')
 labels = learn.dls.vocab
 def predict(img):

 import gradio as gr
 from fastai.vision.all import *
+from huggingface_hub import from_pretrained_fastai
 import skimage
+learner = from_pretrained_fastai("espejelomar/identify-my-cat")
 labels = learn.dls.vocab
 def predict(img):