Update tasks/text.py

tasks/text.py

@@ -4,6 +4,7 @@ from datasets import load_dataset
 from sklearn.metrics import accuracy_score
 import numpy as np
 import random
+import os
 import torch
 from torch.utils.data import Dataset, DataLoader
 from transformers import AutoConfig, AutoModelForSequenceClassification, AutoTokenizer