Update tasks/text.py

tasks/text.py

@@ -3,7 +3,7 @@ from datetime import datetime
 from datasets import load_dataset
 from sklearn.metrics import accuracy_score
 import random
-
+import torch
 from .utils.evaluation import TextEvaluationRequest
 from .utils.emissions import tracker, clean_emissions_data, get_space_info
 from transformers import AutoModelForSequenceClassification,AutoTokenizer