ghosthamlet commited on
Commit
c26d24a
1 Parent(s): d901b1a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -14,11 +14,13 @@ def text_generate(prompt, generated_txt):
14
  json_ = {"inputs": prompt,
15
  "parameters":
16
  {
17
- "top_p": 0.5,
 
18
  "temperature": 1.0,
19
  "max_new_tokens": 250,
20
  "return_full_text": True,
21
  "do_sample":True,
 
22
  },
23
  "options":
24
  {"use_cache": True,
 
14
  json_ = {"inputs": prompt,
15
  "parameters":
16
  {
17
+ "top_p": 0.8,
18
+ "top_k": 100,
19
  "temperature": 1.0,
20
  "max_new_tokens": 250,
21
  "return_full_text": True,
22
  "do_sample":True,
23
+ "num_beams": 3,
24
  },
25
  "options":
26
  {"use_cache": True,