Sengxian commited on
Commit
a1b3fa3
·
1 Parent(s): df3518f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -114,9 +114,9 @@ if __name__ == "__main__":
114
  """
115
  BaseStrategy
116
  """)
117
- temperature = gr.Slider(maximum=1, value=0.6, minimum=0, label='Temperature')
118
  topk = gr.Slider(maximum=40, value=0, minimum=0, step=1, label='Top K')
119
- topp = gr.Slider(maximum=1, value=0.5, minimum=0, label='Top P')
120
 
121
  inputs = [model_input, seed, out_seq_length, min_gen_length, sampling_strategy, num_beams, length_penalty, no_repeat_ngram_size, temperature, topk, topp]
122
  gen.click(fn=predict, inputs=inputs, outputs=outputs)
 
114
  """
115
  BaseStrategy
116
  """)
117
+ temperature = gr.Slider(maximum=1, value=1.0, minimum=0, label='Temperature')
118
  topk = gr.Slider(maximum=40, value=0, minimum=0, step=1, label='Top K')
119
+ topp = gr.Slider(maximum=1, value=0.7, minimum=0, label='Top P')
120
 
121
  inputs = [model_input, seed, out_seq_length, min_gen_length, sampling_strategy, num_beams, length_penalty, no_repeat_ngram_size, temperature, topk, topp]
122
  gen.click(fn=predict, inputs=inputs, outputs=outputs)