Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -23,7 +23,7 @@ model, tokenizer = load_model_and_tokenizer()
|
|
23 |
|
24 |
|
25 |
def get_predict(title: str, abstract: str) -> (str, float, dict):
|
26 |
-
tokenized_text = tokenizer(title + tokenizer.sep_token +
|
27 |
|
28 |
with torch.no_grad():
|
29 |
outputs = model(tokenized_text)
|
|
|
23 |
|
24 |
|
25 |
def get_predict(title: str, abstract: str) -> (str, float, dict):
|
26 |
+
tokenized_text = tokenizer(title + tokenizer.sep_token + abstract[:128], padding="max_length", truncation=True)
|
27 |
|
28 |
with torch.no_grad():
|
29 |
outputs = model(tokenized_text)
|