Sengxian commited on
Commit
7b6c79a
1 Parent(s): dc9694c

Update app.py

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