Hugging Face's logo

Spaces:
Sa-m
/
Brand-Logo-Classification
Runtime error

App Files Community
Sa-m commited on
Commit
6040ac9
·
1 Parent(s): e3408df

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -8,7 +8,11 @@ from tensorflow.keras.applications.inception_resnet_v2 import InceptionResNetV2
8
 
9
  #os.environ['TF_KERAS'] = '1'
10
  #os.environ['TF_ENABLE_ONEDNN_OPTS']='0'
 
 
 
11
  model=load_model('best_model.h5')
 
12
 
13
 
14
  def classify_image(inp):
 
8
 
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
 
18
  def classify_image(inp):