Update tasks/image.py

tasks/image.py

@@ -173,7 +173,6 @@ async def evaluate_image(request: ImageEvaluationRequest):
     ])
 
     # Dataset Loader for CNN computation
-    train_loader = DataLoader(train_test, batch_size=64, shuffle=False)
     test_loader = DataLoader(test_dataset, batch_size=64, shuffle=True)
 
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")