aewurm98 commited on
Commit
f29bed0
1 Parent(s): 0b105a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -1
app.py CHANGED
@@ -13,7 +13,16 @@ import gradio as gr
13
  learn_inf = load_learner('export2.pkl')
14
 
15
  # %% ../dog-breeds.ipynb 15
16
- categories = dls.vocab
 
 
 
 
 
 
 
 
 
17
 
18
  def classify_image(img):
19
  pred, idx, probs = learn_inf.predict(img)
 
13
  learn_inf = load_learner('export2.pkl')
14
 
15
  # %% ../dog-breeds.ipynb 15
16
+ categories = ['Abyssinian', 'Bengal', 'Birman', 'Bombay', 'British_Shorthair',
17
+ 'Egyptian_Mau', 'Maine_Coon', 'Persian', 'Ragdoll', 'Russian_Blue',
18
+ 'Siamese', 'Sphynx', 'american_bulldog', 'american_pit_bull_terrier',
19
+ 'basset_hound', 'beagle', 'boxer', 'chihuahua', 'english_cocker_spaniel',
20
+ 'english_setter', 'german_shorthaired', 'great_pyrenees', 'havanese',
21
+ 'japanese_chin', 'keeshond', 'leonberger', 'miniature_pinscher',
22
+ 'newfoundland', 'pomeranian', 'pug', 'saint_bernard', 'samoyed',
23
+ 'scottish_terrier', 'shiba_inu', 'staffordshire_bull_terrier',
24
+ 'wheaten_terrier', 'yorkshire_terrier']
25
+
26
 
27
  def classify_image(img):
28
  pred, idx, probs = learn_inf.predict(img)