Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -112,9 +112,9 @@ if __name__ == "__main__":
|
|
112 |
"""
|
113 |
BaseStrategy
|
114 |
""")
|
115 |
-
temperature = gr.Slider(maximum=1, value=0.
|
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)
|