Update app.py
Browse files
app.py
CHANGED
|
@@ -270,9 +270,9 @@ def Generate_Karaoke(input_lyrics,
|
|
| 270 |
with ctx:
|
| 271 |
out = kar_model.generate(x,
|
| 272 |
672,
|
| 273 |
-
temperature=
|
| 274 |
-
filter_logits_fn=
|
| 275 |
-
filter_kwargs={'
|
| 276 |
return_prime=False,
|
| 277 |
eos_token=20386,
|
| 278 |
verbose=True)
|
|
@@ -445,7 +445,7 @@ with gr.Blocks() as demo:
|
|
| 445 |
gr.Markdown("## Generation options")
|
| 446 |
|
| 447 |
model_temperature = gr.Slider(0.1, 1, value=0.9, step=0.01, label="Model temperature")
|
| 448 |
-
model_sampling_top_k = gr.Slider(1, 100, value=
|
| 449 |
|
| 450 |
generate_btn = gr.Button("Generate", variant="primary")
|
| 451 |
|
|
|
|
| 270 |
with ctx:
|
| 271 |
out = kar_model.generate(x,
|
| 272 |
672,
|
| 273 |
+
temperature=model_temperature,
|
| 274 |
+
filter_logits_fn=top_k,
|
| 275 |
+
filter_kwargs={'k': model_sampling_top_k},
|
| 276 |
return_prime=False,
|
| 277 |
eos_token=20386,
|
| 278 |
verbose=True)
|
|
|
|
| 445 |
gr.Markdown("## Generation options")
|
| 446 |
|
| 447 |
model_temperature = gr.Slider(0.1, 1, value=0.9, step=0.01, label="Model temperature")
|
| 448 |
+
model_sampling_top_k = gr.Slider(1, 100, value=5, step=1, label="Model sampling top k value")
|
| 449 |
|
| 450 |
generate_btn = gr.Button("Generate", variant="primary")
|
| 451 |
|