Spaces:

sl207
/

neatclf

Sleeping

sl207 commited on May 28

Commit

3453836

•

1 Parent(s): ba7f700

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ from fastai.vision.all import *
 import skimage
 learn = load_learner('model.pkl')
-labels = ['Adidas Shoe', 'New Balance Shoe', 'Nike Show']
 def predict(image):
     image = PILImage.create(image)
     brand, idx, probs = learn.predict(image)

 import skimage
 learn = load_learner('model.pkl')
+labels = learn.dls.vocab
 def predict(image):
     image = PILImage.create(image)
     brand, idx, probs = learn.predict(image)