fix the import names and the name of the variables in list comprehension

app.py

@@ -1,5 +1,5 @@
 ### 1. Imports and class names setup ###
-from model import create_vit_model
+from model import create_vitB16_model
 
 import torch
 from typing import Tuple, Dict
@@ -9,11 +9,11 @@ import os
 
 # Open Food101 class names file and read each line into a list
 with open('class_names.txt', 'r') as f:
-  class_names = [food.strip() for food in food101_class_names]
+  class_names = [food.strip() for food in f.readlines()]
 
 
-### 2. Model and transforms perparation ###
-model, model_transforms = create_vitB16_model(num_classes=len(class_names))
+### 2. Model and transforms preparation ###
+model, model_transforms = create_vitB16_model(num_classes=101)
 
 # Load save weights
 model.load_state_dict(torch.load(f='09_pretrained_vit_feature_extractor_food101_20_percent.pth',