Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -20,7 +20,7 @@ def generate_text(input_text):
|
|
| 20 |
input_ids,
|
| 21 |
attention_mask=attention_mask,
|
| 22 |
max_length=200,
|
| 23 |
-
max_new_tokens=400, top_p = 0.7, top_k = 50
|
| 24 |
do_sample=True,
|
| 25 |
num_return_sequences=1,
|
| 26 |
eos_token_id=tokenizer.eos_token_id,
|
|
|
|
| 20 |
input_ids,
|
| 21 |
attention_mask=attention_mask,
|
| 22 |
max_length=200,
|
| 23 |
+
max_new_tokens=400, top_p = 0.7, top_k = 50,
|
| 24 |
do_sample=True,
|
| 25 |
num_return_sequences=1,
|
| 26 |
eos_token_id=tokenizer.eos_token_id,
|