Nitzantry1 commited on
Commit
1c8f97a
·
verified ·
1 Parent(s): 5c7f79f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=1000,
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)