aagirre92 commited on
Commit
de0d79c
1 Parent(s): 824c00e

wrong path directory in class_names.txt

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ import torchvision
15
  # Setup class names (hardcoded, these shall reside in a json file or sth like that...)
16
  # Open Food101 class names fromt file and import it to list
17
 
18
- with open("demos/foodvision_big/class_names.txt","r") as f:
19
  class_names = f.read().split("\n")
20
 
21
  ### 2. Model and transforms preparation ###
 
15
  # Setup class names (hardcoded, these shall reside in a json file or sth like that...)
16
  # Open Food101 class names fromt file and import it to list
17
 
18
+ with open("class_names.txt","r") as f:
19
  class_names = f.read().split("\n")
20
 
21
  ### 2. Model and transforms preparation ###