Update tasks/text.py

tasks/text.py

@@ -4,6 +4,7 @@ from datasets import load_dataset
 from torch.utils.data import DataLoader
 from sklearn.metrics import accuracy_score
 import random
+import os
 
 from .utils.evaluation import TextEvaluationRequest
 from .utils.emissions import tracker, clean_emissions_data, get_space_info