Spaces:

MatthiasPicard
/

submission-qwen

Sleeping

Rcarvalo commited on Jan 23

Commit

822b9e4

verified ·

1 Parent(s): 5e2b367

Update tasks/text.py

Files changed (1) hide show

tasks/text.py CHANGED Viewed

@@ -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
 router = APIRouter()

 import torch
 from .utils.evaluation import TextEvaluationRequest
 from .utils.emissions import tracker, clean_emissions_data, get_space_info
+from transformers import AutoTokenizer,BertForSequenceClassification,AutoModelForSequenceClassification,Trainer, TrainingArguments,DataCollatorWithPadding
 router = APIRouter()