Update tasks/text.py

tasks/text.py

@@ -66,7 +66,7 @@ def bert_classifier(test_dataset: dict, model: str):
         raise(ValueError)
 
     # Use CUDA if available 
-    device, _, _ = get_backend()
+    device = "cuda"
 
     model = model.to(device)