Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ import os
|
|
6 |
# Configuration
|
7 |
HEIGHT, WIDTH = 224, 224
|
8 |
NUM_CLASSES = 6
|
9 |
-
LABELS = ["Burger King",
|
10 |
from tensorflow_addons.metrics import F1Score
|
11 |
from keras.utils import custom_object_scope
|
12 |
|
|
|
6 |
# Configuration
|
7 |
HEIGHT, WIDTH = 224, 224
|
8 |
NUM_CLASSES = 6
|
9 |
+
LABELS = [ "McDonalds","Burger King","Subway", "Starbucks", "KFC","Other"]
|
10 |
from tensorflow_addons.metrics import F1Score
|
11 |
from keras.utils import custom_object_scope
|
12 |
|