Sa-m commited on
Commit
f50b1a5
1 Parent(s): a077d32

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ import os
9
  #os.environ['TF_KERAS'] = '1'
10
  #os.environ['TF_ENABLE_ONEDNN_OPTS']='0'
11
 
12
- labels=['Subway', 'Starbucks', 'McDonalds', 'Burger King', 'KFC', 'Other']
13
  HEIGHT,WIDTH=224,224
14
  model=load_model('best_model2.h5')
15
  NUM_CLASSES=6
 
9
  #os.environ['TF_KERAS'] = '1'
10
  #os.environ['TF_ENABLE_ONEDNN_OPTS']='0'
11
 
12
+ labels={'Subway': 0, 'Starbucks': 1,'McDonalds': 2,'Burger King': 3,'KFC': 4,'Other': 5}
13
  HEIGHT,WIDTH=224,224
14
  model=load_model('best_model2.h5')
15
  NUM_CLASSES=6