Spaces:

Sa-m
/

Brand-Logo-Classification

Runtime error

Sa-m commited on Sep 10, 2022

Commit

5e4e204

•

1 Parent(s): a1f1908

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ import os
 #os.environ['TF_KERAS'] = '1'
 #os.environ['TF_ENABLE_ONEDNN_OPTS']='0'
-labels=['Other', 'Starbucks', 'Subway', 'McDonalds', 'KFC', 'Burger King']
 HEIGHT,WIDTH=224,224
 model=load_model('best_model2.h5')
 NUM_CLASSES=6

 #os.environ['TF_KERAS'] = '1'
 #os.environ['TF_ENABLE_ONEDNN_OPTS']='0'
+labels=['Subway', 'Starbucks', 'McDonalds', 'Burger King', 'KFC', 'Other']
 HEIGHT,WIDTH=224,224
 model=load_model('best_model2.h5')
 NUM_CLASSES=6