BigSalmon commited on
Commit
a0ed721
·
1 Parent(s): 2a2f758

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -27,6 +27,7 @@ def extend(input_text, num_return_sequences, bad_words, max_size=20, top_k=50, t
27
  else:
28
  input_ids = encoded_prompt
29
  bad_words = bad_words.split()
 
30
  bad_word_ids = []
31
  for bad_word in bad_words:
32
  bad_word = " " + bad_word
 
27
  else:
28
  input_ids = encoded_prompt
29
  bad_words = bad_words.split()
30
+ print(bad_words)
31
  bad_word_ids = []
32
  for bad_word in bad_words:
33
  bad_word = " " + bad_word