asigalov61 commited on
Commit
151dbdb
·
verified ·
1 Parent(s): 8aaccd6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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=0.85,
274
- filter_logits_fn=top_p,
275
- filter_kwargs={'thres': 0.96},
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=15, step=1, label="Model sampling top k 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