Spaces:
Sleeping
Sleeping
Update tasks/text.py
Browse files- tasks/text.py +2 -2
tasks/text.py
CHANGED
@@ -72,8 +72,8 @@ async def evaluate_text(request: TextEvaluationRequest):
|
|
72 |
tokenizer = AutoTokenizer.from_pretrained(path_tokenizer)
|
73 |
|
74 |
def preprocess_function(df):
|
75 |
-
return tokenizer(df["quote"], truncation=True)
|
76 |
-
tokenized_test = test_dataset.map(preprocess_function, batched=True
|
77 |
|
78 |
# training_args = torch.load("training_args.bin")
|
79 |
# training_args.eval_strategy='no'
|
|
|
72 |
tokenizer = AutoTokenizer.from_pretrained(path_tokenizer)
|
73 |
|
74 |
def preprocess_function(df):
|
75 |
+
return tokenizer(df["quote"], truncation=True,padding='longest')
|
76 |
+
tokenized_test = test_dataset.map(preprocess_function, batched=True)
|
77 |
|
78 |
# training_args = torch.load("training_args.bin")
|
79 |
# training_args.eval_strategy='no'
|