Update app.py

app.py

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