Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -88,7 +88,7 @@ Extract explicit and implicit arguments from historical text.
|
|
| 88 |
{text}<|eot_id|>
|
| 89 |
<|start_header_id|>assistant<|end_header_id|>"""
|
| 90 |
|
| 91 |
-
inputs = tokenizer(prompt, return_tensors="pt", max_length=5048).to(model.device)
|
| 92 |
input_length = inputs['input_ids'].shape[1]
|
| 93 |
try:
|
| 94 |
if temperature is None:
|
|
|
|
| 88 |
{text}<|eot_id|>
|
| 89 |
<|start_header_id|>assistant<|end_header_id|>"""
|
| 90 |
|
| 91 |
+
inputs = tokenizer(prompt, return_tensors="pt",truncation=True, max_length=5048).to(model.device)
|
| 92 |
input_length = inputs['input_ids'].shape[1]
|
| 93 |
try:
|
| 94 |
if temperature is None:
|