Upload app.py

app.py

@@ -9,7 +9,7 @@ from typing import Tuple, Dict
 
 # Setup class names
 with open("class_names.txt", "r") as f:
-    class_names = [food_name.strip() for food_name in f.readline()]
+    class_names = [food_name.strip() for food_name in f.readlines()]
 
 effnetb2, effnetb2_transforms = create_effnetb2_model(
     num_classes=101
@@ -50,7 +50,7 @@ demo = gr.Interface(
     inputs=gr.Image(type="pil"),
     outputs=[
         gr.Label(num_top_classes=5, label="Predictions"),
-        gr.Number(label="Prdiction time(s)"),
+        gr.Number(label="Prediction time(s)"),
     ],
     examples=example_list,
     title=title,