Update tasks/text.py

tasks/text.py

@@ -28,15 +28,10 @@ if torch.cuda.is_available():
     device = torch.device("cuda")
 else:
     device = torch.device("cpu")
+print(device)
 
 
-if torch.cuda.is_available():
-    device = torch.device("cuda")
-else:
-    device = torch.device("cpu")
-
-
-MODEL = "modern-large" #mlp, ct, modern-base, modern-large, gte-base, gte-large
+MODEL = "modern-base" #mlp, ct, modern-base, modern-large, gte-base, gte-large
 
 class ConspiracyClassification(
     nn.Module,