Rishikesh22 commited on
Commit
362b6e1
1 Parent(s): fdcbb34

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -6,6 +6,9 @@ from model import create_effnetb2_model
6
  from timeit import default_timer as timer
7
  from typing import Tuple, Dict
8
 
 
 
 
9
  # Setup class names
10
  with open("class_names.txt", "r") as f: # reading them in from class_names.txt
11
  class_names = [food_name.strip() for food_name in f.readlines()]
 
6
  from timeit import default_timer as timer
7
  from typing import Tuple, Dict
8
 
9
+ # Add a line at the top
10
+ print("Check out the list of classes [here](https://huggingface.co/spaces/Rishikesh22/Food_classification/raw/main/class_names.txt)")
11
+
12
  # Setup class names
13
  with open("class_names.txt", "r") as f: # reading them in from class_names.txt
14
  class_names = [food_name.strip() for food_name in f.readlines()]