Spaces:
Running
Running
Nitzantry1
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -13,7 +13,7 @@ def generate_text(prompt):
|
|
13 |
attention_mask=inputs.attention_mask,
|
14 |
do_sample=True,
|
15 |
temperature=0.9,
|
16 |
-
max_length=
|
17 |
pad_token_id=tokenizer.eos_token_id
|
18 |
)
|
19 |
gen_text = tokenizer.decode(gen_tokens[0], skip_special_tokens=True)
|
|
|
13 |
attention_mask=inputs.attention_mask,
|
14 |
do_sample=True,
|
15 |
temperature=0.9,
|
16 |
+
max_length=100,
|
17 |
pad_token_id=tokenizer.eos_token_id
|
18 |
)
|
19 |
gen_text = tokenizer.decode(gen_tokens[0], skip_special_tokens=True)
|