Update tasks/text.py

tasks/text.py

@@ -6,7 +6,7 @@ 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
+from transformers import AutoTokenizer,BertForSequenceClassification,AutoModelForSequenceClassification,Trainer, TrainingArguments,DataCollatorWithPadding
 
 router = APIRouter()