PyaeSoneK commited on
Commit
3fedd51
·
1 Parent(s): cb5e41b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +36,7 @@ pipe = pipeline("text-generation",
36
  max_new_tokens = 512,
37
  do_sample=True,
38
  top_k=30,
39
- num_return_sequences=4,
40
  eos_token_id=tokenizer.eos_token_id
41
  )
42
 
 
36
  max_new_tokens = 512,
37
  do_sample=True,
38
  top_k=30,
39
+ num_return_sequences=2,
40
  eos_token_id=tokenizer.eos_token_id
41
  )
42