Spaces:
Running
Running
aliasgerovs
commited on
Commit
•
414fead
1
Parent(s):
4451e36
Updated
Browse files
app.py
CHANGED
@@ -269,7 +269,7 @@ def split_text_allow_complete_sentences_nltk(text, max_length=256, tolerance=30,
|
|
269 |
|
270 |
def predict_bc(model, tokenizer, text):
|
271 |
tokens = text_bc_tokenizer(
|
272 |
-
text, padding='max_length', truncation=True, max_length=
|
273 |
).to(device)["input_ids"]
|
274 |
|
275 |
output = model(tokens)
|
|
|
269 |
|
270 |
def predict_bc(model, tokenizer, text):
|
271 |
tokens = text_bc_tokenizer(
|
272 |
+
text, padding='max_length', truncation=True, max_length=256, return_tensors="pt"
|
273 |
).to(device)["input_ids"]
|
274 |
|
275 |
output = model(tokens)
|