This commit is contained in:
Maciej Ścigacz 2023-06-14 00:42:35 +02:00
parent ddb3326084
commit f36610e009

View File

@ -14,7 +14,6 @@ def clear_data(data):
return data return data
def predict_labels(texts): def predict_labels(texts):
encodings = tokenizer(texts, truncation=True, padding=True, return_tensors='pt') encodings = tokenizer(texts, truncation=True, padding=True, return_tensors='pt')
input_ids = encodings['input_ids'] input_ids = encodings['input_ids']