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)
|