Hugging Face's logo

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

App Files Community
Sa-m commited on
Commit
68a11e7
·
1 Parent(s): 107d5a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import tensorflow_addons as tfa
7
  import os
8
  from tensorflow.keras.layers import *
9
 
10
- labels={'Subway': 0, 'Starbucks': 1,'McDonalds': 2,'Burger King': 3,'KFC': 4,'Other': 5}
11
  HEIGHT,WIDTH=224,224
12
  model=load_model('best_model2.h5')
13
  NUM_CLASSES=6
 
7
  import os
8
  from tensorflow.keras.layers import *
9
 
10
+ labels= {'Subway': 0, 'Starbucks': 1,'McDonalds': 2,'Burger King': 3,'KFC': 4,'Other': 5}
11
  HEIGHT,WIDTH=224,224
12
  model=load_model('best_model2.h5')
13
  NUM_CLASSES=6