Hugging Face's logo

Spaces:
Sa-m
/
Brand-Logo-Classification
Running

App Files
Sa-m commited on
Commit
65cd9a4
•
1 Parent(s): 81aa00c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ model=load_model('best_model2.h5')
31
 
32
  def classify_image(inp):
33
  np.random.seed(143)
34
- labels = {'Burger King': 0, 'KFC': 1, 'McDonalds': 2, 'Other': 3, 'Starbucks': 4, 'Subway': 5}
35
  NUM_CLASSES = 6
36
  inp = inp.reshape((-1, HEIGHT, WIDTH, 3))
37
  inp = tf.keras.applications.nasnet.preprocess_input(inp)
 
31
 
32
  def classify_image(inp):
33
  np.random.seed(143)
34
+ labels = {'Burger King': 1, 'KFC': 0, 'McDonalds': 2, 'Other': 3, 'Starbucks': 4, 'Subway': 5}
35
  NUM_CLASSES = 6
36
  inp = inp.reshape((-1, HEIGHT, WIDTH, 3))
37
  inp = tf.keras.applications.nasnet.preprocess_input(inp)