aliasgerovs commited on
Commit
b770251
1 Parent(s): 6bc69b2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -266,7 +266,7 @@ def predict_bc(model, tokenizer, text):
266
 
267
  def predict_mc(model, tokenizer, text):
268
  tokens = tokenizer(
269
- text, padding='max_length', truncation=True, return_tensors="pt", max_length=512
270
  ).to(device)["input_ids"]
271
  output = model(tokens)
272
  output_norm = softmax(output.logits.detach().cpu().numpy(), 1)[0]
 
266
 
267
  def predict_mc(model, tokenizer, text):
268
  tokens = tokenizer(
269
+ text, padding='max_length', truncation=True, return_tensors="pt", max_length=256
270
  ).to(device)["input_ids"]
271
  output = model(tokens)
272
  output_norm = softmax(output.logits.detach().cpu().numpy(), 1)[0]