Jofthomas HF staff commited on
Commit
b7323e4
·
verified ·
1 Parent(s): 69cd7c3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -421,7 +421,11 @@ def generate_beams(n_beams, start_sentence, scores, length_penalty, decoded_sequ
421
 
422
  @spaces.GPU
423
  def get_beam_search_html(
424
- input_text, number_steps, number_beams, length_penalty, num_return_sequences
 
 
 
 
425
  ):
426
  inputs = tokenizer([input_text], return_tensors="pt")
427
 
 
421
 
422
  @spaces.GPU
423
  def get_beam_search_html(
424
+ input_text,
425
+ number_steps=12,
426
+ number_beams=1,
427
+ length_penalty=1.0,
428
+ num_return_sequences=3
429
  ):
430
  inputs = tokenizer([input_text], return_tensors="pt")
431