a

.ipynb_checkpoints/app-checkpoint.py

@@ -8,7 +8,7 @@ from torchvision.datasets import ImageFolder
 from transformers import ViTFeatureExtractor, ViTForImageClassification,  AutoFeatureExtractor, ViTMSNForImageClassification
 from pathlib import Path
 import pytorch_lightning as pl
-
+from torch.utils.data import DataLoader
 
 def video_identity(video,user_name,class_name,trainortest,ready):
     if ready=='yes':

app.py

@@ -8,7 +8,7 @@ from torchvision.datasets import ImageFolder
 from transformers import ViTFeatureExtractor, ViTForImageClassification,  AutoFeatureExtractor, ViTMSNForImageClassification
 from pathlib import Path
 import pytorch_lightning as pl
-
+from torch.utils.data import DataLoader
 
 def video_identity(video,user_name,class_name,trainortest,ready):
     if ready=='yes':