Spaces:

Sa-m
/

Brand-Logo-Classification

Runtime error

Sa-m commited on Sep 11, 2022

Commit

4f141df

•

1 Parent(s): 68a11e7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ from tensorflow.keras.layers import *
 labels= {'Subway': 0, 'Starbucks': 1,'McDonalds': 2,'Burger King': 3,'KFC': 4,'Other': 5}
 HEIGHT,WIDTH=224,224
-model=load_model('best_model2.h5')
 NUM_CLASSES=6

 labels= {'Subway': 0, 'Starbucks': 1,'McDonalds': 2,'Burger King': 3,'KFC': 4,'Other': 5}
 HEIGHT,WIDTH=224,224
+model=load_model('best_model.h5')
 NUM_CLASSES=6