m-ric HF staff commited on
Commit
c4a1c93
β€’
1 Parent(s): 0033238

Update app.py

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