Spaces:

Sa-m
/

Brand-Logo-Classification

Runtime error

Sa-m commited on Sep 10, 2022

Commit

e13a8d8

•

1 Parent(s): 36697f7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,7 +11,7 @@ import os
 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