hanyullai commited on
Commit
a47f8a0
1 Parent(s): bd9c730

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ def predict(text, seed, out_seq_length, min_gen_length, sampling_strategy,
28
  "sampling_strategy": sampling_strategy,
29
  "num_beams": num_beams,
30
  "max_tokens": out_seq_length,
31
- "no_repeat_ngram": no_repeat_ngram,
32
  "seed": seed
33
  })
34
 
 
28
  "sampling_strategy": sampling_strategy,
29
  "num_beams": num_beams,
30
  "max_tokens": out_seq_length,
31
+ "no_repeat_ngram": no_repeat_ngram_size,
32
  "seed": seed
33
  })
34