Tonic commited on
Commit
d49b434
1 Parent(s): c0af598

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ def generate_code(prompt):
24
  **inputs,
25
  max_new_tokens=650,
26
  temperature=0.3,
27
- do_sample=False,
28
  )
29
  generated_code = tokenizer.decode(tokens[0], skip_special_tokens=True)
30
  return generated_code
 
24
  **inputs,
25
  max_new_tokens=650,
26
  temperature=0.3,
27
+ do_sample=True,
28
  )
29
  generated_code = tokenizer.decode(tokens[0], skip_special_tokens=True)
30
  return generated_code