Spaces:

Frantz103
/

cat_dogs

Sleeping

Frantz103 commited on Aug 15, 2023

Commit

38921ae

•

1 Parent(s): f2e88cc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ def is_cat(x): return x[0].isupper()
 learn_inf = load_learner('DogCat.pkl')
 labels = learn_inf.dls.vocab
-labels = ['Dog', 'Cat']
 def predict(img):
     img = PILImage.create(img)

 learn_inf = load_learner('DogCat.pkl')
 labels = learn_inf.dls.vocab
+labels = ['Cat', 'Not_cat']
 def predict(img):
     img = PILImage.create(img)