Spaces:
Running
Running
aliasgerovs
commited on
Commit
•
b770251
1
Parent(s):
6bc69b2
Update app.py
Browse files
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=
|
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]
|