mayank2 commited on
Commit
81ab2e5
1 Parent(s): 9d876d1

added changes

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. export.pkl +2 -2
app.py CHANGED
@@ -3,8 +3,8 @@ from fastai.vision.all import *
3
  import skimage
4
 
5
  learn = load_learner('export.pkl')
6
- def is_cat(x): return x[0].isupper()
7
- labels = is_cat
8
  def predict(img):
9
  img = PILImage.create(img)
10
  pred,pred_idx,probs = learn.predict(img)
 
3
  import skimage
4
 
5
  learn = load_learner('export.pkl')
6
+
7
+ labels = learn.dls.vocab
8
  def predict(img):
9
  img = PILImage.create(img)
10
  pred,pred_idx,probs = learn.predict(img)
export.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:aaf77d8bea0e214dc4751fb7d8514050a220452b28a04d6668b7a2b618349e76
3
- size 47062507
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d29df3511ce532fa048454834383e5790cd3575b7fd50b18c0944d0e7fde529c
3
+ size 103043993