Update tasks/text.py

tasks/text.py

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