Loewolf commited on
Commit
b24ef2a
1 Parent(s): 433119f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,10 +16,10 @@ def generate_text(prompt):
16
  attention_mask=attention_mask,
17
  max_length=max_length,
18
  min_length=4,
19
- num_beams=20,
20
  no_repeat_ngram_size=2,
21
  early_stopping=True,
22
- temperature=0.6,
23
  top_p=0.95,
24
  top_k=70,
25
  length_penalty=2.0,
 
16
  attention_mask=attention_mask,
17
  max_length=max_length,
18
  min_length=4,
19
+ num_beams=5,
20
  no_repeat_ngram_size=2,
21
  early_stopping=True,
22
+ temperature=0.7,
23
  top_p=0.95,
24
  top_k=70,
25
  length_penalty=2.0,