anas-awadalla commited on
Commit
45883da
·
1 Parent(s): 6f69561

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -158,6 +158,9 @@ def generate(
158
  attention_mask=attention_mask,
159
  max_new_tokens=30,
160
  num_beams=3,
 
 
 
161
  )
162
 
163
  gen_text = tokenizer.decode(
 
158
  attention_mask=attention_mask,
159
  max_new_tokens=30,
160
  num_beams=3,
161
+ do_sample=True,
162
+ temperature=0.7,
163
+ top_k=0,
164
  )
165
 
166
  gen_text = tokenizer.decode(