Update app.py

app.py

@@ -9,11 +9,11 @@ from transformers import ViTForImageClassification, AutoImageProcessor, AdamW, V
 from torch.utils.data import DataLoader, TensorDataset
 import gradio as gr
 
-model_path = '/home/usr/app'
+model_path = '/home/user/app'
 train_pickle_path = 'train_data.pickle'
 valid_pickle_path = 'valid_data.pickle'
 image_directory = 'images' 
-test_image_path = '/home/usr/app/test.jpg'
+test_image_path = '/home/user/app/test.jpg'
 num_epochs = 5    # Fine-tune the model
 label_list = ["小白", "巧巧", "冏媽", "乖狗", "花捲", "超人", "黑胖", "橘子"]
 label_dictionary = {"小白": 0, "巧巧": 1, "冏媽": 2, "乖狗": 3, "花捲": 4, "超人": 5, "黑胖": 6, "橘子": 7}