a

app.py

@@ -76,8 +76,8 @@ def video_identity(video,user_name,class_name,trainortest,ready):
 
             
            
-        train_loader = DataLoader(train_ds, batch_size=1, collate_fn=collator, num_workers=0, shuffle=True)
-        test_loader = DataLoader(test_ds, batch_size=1, collate_fn=collator, num_workers=0)
+        train_loader = DataLoader(train_ds, batch_size=1, collate_fn=collator, num_workers=8, shuffle=True)
+        test_loader = DataLoader(test_ds, batch_size=1, collate_fn=collator, num_workers=8)
         
         
         for name, param in model.named_parameters():