fix the app.py

app.py

@@ -7,18 +7,19 @@ from timeit import default_timer as timer
 import gradio as gr
 import os
 
+import 
+
 class_names = []
 
-# Open Food101 class names file and read each line into a list
-with open(class_names_file, 'r') as f:
-    for idx, class_name in enumerate(f):
-        class_names[idx] = class_name.strip() 
+# Open Cifar class names file and read each line into a list
+with open('class_names.txt', 'r') as f:
+    class_names = [obj.strip() for obj in cifar_class_names]
 
-### 2. Model and transforms perparation ###
+### 2. Model and transforms preparation ###
 model, model_transforms = create_vitB16_model(num_classes=len(class_names))
 
 # Load save weights
-model.load_state_dict(torch.load(f='09_pretrained_vit_feature_extractor_food101_20_percent.pth',
+model.load_state_dict(torch.load(f='09_pretrained_vit_feature_extractor_cifar_60_percent.pth',
                                  map_location='cpu'))
 
 # 3. Predict Function