aliasgerovs commited on
Commit
414fead
1 Parent(s): 4451e36
Files changed (1) hide show
  1. app.py +1 -1
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=512, return_tensors="pt"
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)