Spaces:
Sleeping
Sleeping
Adjusted app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ from model import create_effnetb2_model
|
|
8 |
from timeit import default_timer as timer
|
9 |
|
10 |
# Setup class names
|
11 |
-
with open('
|
12 |
class_names = [food_name.strip() for food_name in f.readlines()]
|
13 |
|
14 |
### 2. Model and transforms preparation ###
|
|
|
8 |
from timeit import default_timer as timer
|
9 |
|
10 |
# Setup class names
|
11 |
+
with open('demos/foodvision_big/class_names.txt', 'r') as f:
|
12 |
class_names = [food_name.strip() for food_name in f.readlines()]
|
13 |
|
14 |
### 2. Model and transforms preparation ###
|