m-ric HF staff commited on
Commit
9ff548f
β€’
1 Parent(s): bf09de3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -429,7 +429,7 @@ def get_beam_search_html(
429
  **inputs,
430
  max_new_tokens=number_steps,
431
  num_beams=number_beams,
432
- num_return_sequences=1,
433
  return_dict_in_generate=True,
434
  length_penalty=length_penalty,
435
  output_scores=True,
 
429
  **inputs,
430
  max_new_tokens=number_steps,
431
  num_beams=number_beams,
432
+ num_return_sequences=num_return_sequences,
433
  return_dict_in_generate=True,
434
  length_penalty=length_penalty,
435
  output_scores=True,