Update app.py

app.py

@@ -9,7 +9,7 @@ import torch
 import os
 import gradio as gr 
 
-with open("demo/class_names.txt", 'r') as file:
+with open("class_names.txt", 'r') as file:
     class_names = [line.strip() for line in file]
 
 
@@ -18,7 +18,7 @@ model, transforms = vit_b_16(num_classes=101,
                              freeze_gradients=True,
                              unfreeze_blocks=0)
 
-model.load_state_dict(torch.load('demo/vit_b_16_unfreeze_one_encoder_block_10_total_epochs.pth',
+model.load_state_dict(torch.load('vit_b_16_unfreeze_one_encoder_block_10_total_epochs.pth',
                                   weights_only=True))
 
 
@@ -55,7 +55,7 @@ and fine-tuning the results for specific food categories.
 This model achieved a Top-1 accuracy of 91.55% and a Top-5 accuracy of 98.56%
 """
 
-sample_list = [["demo/samples/" + sample] for sample in os.listdir("demo/samples")]
+sample_list = [["samples/" + sample] for sample in os.listdir("samples")]
 
 #Gradio interface
 demo = gr.Interface(