Spaces:
Running
Running
Update app.py
Browse files
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=
|
74 |
-
topp = gr.Slider(maximum=
|
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)
|