Sa-m commited on
Commit
5e4e204
1 Parent(s): a1f1908

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=['Other', 'Starbucks', 'Subway', 'McDonalds', 'KFC', 'Burger King']
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', 'Starbucks', 'McDonalds', 'Burger King', 'KFC', 'Other']
13
  HEIGHT,WIDTH=224,224
14
  model=load_model('best_model2.h5')
15
  NUM_CLASSES=6