hanyullai commited on
Commit
d0186a4
1 Parent(s): db6defc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -70,8 +70,8 @@ if __name__ == "__main__":
70
  with gr.TabItem("Base Search Parameter"):
71
  # base search
72
  temperature = gr.Slider(maximum=1, value=1, minimum=0, label='Temperature')
73
- topk = gr.Slider(maximum=8, value=1, minimum=1, step=1, label='Top K')
74
- topp = gr.Slider(maximum=8, value=0, minimum=0, step=1, label='Top P')
75
 
76
  inputs = [model_input, seed, out_seq_length, min_gen_length, sampling_strategy, num_beams, length_penalty, no_repeat_ngram_size, temperature, topk, topp]
77
  gen.click(fn=predict, inputs=inputs, outputs=outputs)
 
70
  with gr.TabItem("Base Search Parameter"):
71
  # base search
72
  temperature = gr.Slider(maximum=1, value=1, minimum=0, label='Temperature')
73
+ topk = gr.Slider(maximum=8, value=1, minimum=0, step=1, label='Top K')
74
+ topp = gr.Slider(maximum=1, value=0, minimum=0, label='Top P')
75
 
76
  inputs = [model_input, seed, out_seq_length, min_gen_length, sampling_strategy, num_beams, length_penalty, no_repeat_ngram_size, temperature, topk, topp]
77
  gen.click(fn=predict, inputs=inputs, outputs=outputs)