a

.ipynb_checkpoints/app-checkpoint.py

@@ -11,11 +11,11 @@ from transformers import ViTFeatureExtractor, ViTForImageClassification,  AutoFe
 def video_identity(video,user_name,class_name,trainortest,ready):
     if ready=='yes':
 
-        data_dir = Path('train/'+str(user_name))
+        data_dir = Path(str(user_name)+'/train')
         train_ds = ImageFolder(data_dir)
 
         
-        test_dir = Path('test/'+str(user_name))
+        test_dir = Path(str(user_name)+'/test')
         test_ds = ImageFolder(test_dir)        
         
         label2id = {}

app.py

@@ -11,11 +11,11 @@ from transformers import ViTFeatureExtractor, ViTForImageClassification,  AutoFe
 def video_identity(video,user_name,class_name,trainortest,ready):
     if ready=='yes':
 
-        data_dir = Path('train/'+str(user_name))
+        data_dir = Path(str(user_name)+'/train')
         train_ds = ImageFolder(data_dir)
 
         
-        test_dir = Path('test/'+str(user_name))
+        test_dir = Path(str(user_name)+'/test')
         test_ds = ImageFolder(test_dir)        
         
         label2id = {}