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,
|