BigSalmon commited on
Commit
0deb4cd
·
1 Parent(s): a0ed721

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -38,7 +38,7 @@ def extend(input_text, num_return_sequences, bad_words, max_size=20, top_k=50, t
38
  input_ids=input_ids,
39
  max_length=max_size + len(encoded_prompt[0]),
40
  top_k=top_k,
41
- bad_word_ids = bad_word_ids,
42
  top_p=top_p,
43
  do_sample=True,
44
  num_return_sequences=num_return_sequences)
 
38
  input_ids=input_ids,
39
  max_length=max_size + len(encoded_prompt[0]),
40
  top_k=top_k,
41
+ bad_words_ids = bad_word_ids,
42
  top_p=top_p,
43
  do_sample=True,
44
  num_return_sequences=num_return_sequences)