Changed line of code regarding translation device usage, cuda or cpu

src/task_management.py

@@ -68,8 +68,8 @@ class TaskManager:
         )
 
         # Check if CUDA is available and set the device
-        self.translation_device = (
-            "cpu"  # torch.device("cuda" if torch.cuda.is_available() else "cpu")
+        self.translation_device = torch.device(
+            "cuda" if torch.cuda.is_available() else "cpu"
         )
 
         # Load translation pipeline for model facebook/nllb-200-distilled-1.3B