Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -114,9 +114,9 @@ if __name__ == "__main__":
|
|
114 |
"""
|
115 |
BaseStrategy
|
116 |
""")
|
117 |
-
temperature = gr.Slider(maximum=1, value=0
|
118 |
topk = gr.Slider(maximum=40, value=0, minimum=0, step=1, label='Top K')
|
119 |
-
topp = gr.Slider(maximum=1, value=0.
|
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)
|