Spaces:
Running
Running
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 |
|