jtummon commited on
Commit
b654cb4
1 Parent(s): 2ec5e96

Updating app

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -14,12 +14,12 @@ def is_cat(x):
14
  learn = load_learner('model.pkl')
15
 
16
  # %% ../app.ipynb 5
17
- categories = ('American Bulldog', 'American Pit Bull Terrier', 'Basset Hound',
18
- 'Beagle', 'Boxer', 'Chihuahua', 'English Cocker Spaniel', 'English Setter', 'German Shorthaired',
19
- 'Great Pyrenees', 'Havanese', 'Japanese Chin', 'Keeshond', 'Leonberger', 'Miniature Pinscher', 'Newfoundland',
20
- 'Pomeranian', 'Pug', 'St.Bernard', 'Samyoed', 'Scottish Terrier', 'Shiba Inu', 'Staffordshire Bull Terrier',
21
- 'Wheaten Terrier', 'Yorkshire Terrier', 'Abyssian', 'Bengal', 'Birman', 'Bombay', 'British Shorthair',
22
- 'Egyptian Mau', 'Main Coon', 'Persian', 'Ragdoll', 'Russian Blue', 'Siamese', 'Sphynx')
23
 
24
  def classify_image(img):
25
  pred, idx, probs = learn.predict(img)
 
14
  learn = load_learner('model.pkl')
15
 
16
  # %% ../app.ipynb 5
17
+ categories = ('Abyssian', 'American Bulldog', 'American Pit Bull Terrier', 'Basset Hound', 'Beagle','Bengal',
18
+ 'Birman', 'Bombay', 'Boxer','British Shorthair', 'Chihuahua',
19
+ 'Egyptian Mau', 'English Cocker Spaniel', 'English Setter', 'German Shorthaired', 'Great Pyrenees', 'Havanese', 'Japanese Chin',
20
+ 'Keeshond', 'Leonberger', 'Main Coon', 'Miniature Pinscher', 'Newfoundland', 'Persian', 'Pomeranian', 'Pug', 'Ragdoll', 'Russian Blue',
21
+ 'St.Bernard', 'Samyoed', 'Scottish Terrier', 'Shiba Inu', 'Siamese', 'Sphynx',
22
+ 'Staffordshire Bull Terrier', 'Wheaten Terrier', 'Yorkshire Terrier')
23
 
24
  def classify_image(img):
25
  pred, idx, probs = learn.predict(img)