Sa-m commited on
Commit
9796735
1 Parent(s): 8c161eb

Update app.py

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