image path updated

vit_model_original.py

@@ -87,8 +87,8 @@ if __name__ == "__main__":
 
 
 
-    train_real_folder = 'train/art/real'
-    train_fake_folder = 'train/art/fake'
+    train_real_folder = 'datasets/training_set/real'
+    train_fake_folder = 'datasets/training_set/fake'
 
 
 

vit_model_original_test.py

@@ -73,8 +73,8 @@ if __name__ == "__main__":
     ])
 
     # Load the test dataset
-    test_real_folder = 'test/art/real'
-    test_fake_folder = 'test/art/fake'
+    test_real_folder = 'datasets/test_set/real'
+    test_fake_folder = 'datasets/test_set/fake'
     test_set = labeling(test_real_folder, test_fake_folder)
     test_dataset = CustomDataset(test_set, transform=preprocess)
     test_loader = DataLoader(test_dataset, batch_size=32)