Added debug lines

src/deepeval/base_task.py

@@ -14,7 +14,7 @@ class BaseTask(ABC):
     def __init__(self, dataset_repo, model_name):
         self.dataset_repo = dataset_repo
         self.dataset = self.load_dataset_from_hf()
-        self.device = "cuda:0" if torch.cuda.is_available() else "cpu"
+        self.device = "cuda" if torch.cuda.is_available() else "cpu"
         self.model, self.tokenizer = self.get_cached_model(model_name, self.device)
 
 
@@ -166,7 +166,10 @@ class BaseTask(ABC):
         Define your own loading method if needed.
         :return: Dataset
         """
-        return load_dataset(self.dataset_repo, token=HF_TOKEN, split="train")
+        print("Loading dataset from Hugging Face.")
+        dataset= load_dataset(self.dataset_repo, token=HF_TOKEN, split="train")
+        print("Dataset loaded.")
+        return dataset
 
     @abstractmethod
     def evaluate(self):