BigSalmon commited on
Commit
0d39a4d
1 Parent(s): b4c807e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ def run_generate(bad_words):
18
  bad_word = " " + bad_word
19
  ids = tokenizer(bad_word).input_ids
20
  ids = str(ids)
21
- ids = ids.replace("]", ": -30").replace("[", "")
22
  bad_word_ids.append(ids)
23
  bad_word_ids = str(bad_word_ids)
24
  bad_word_ids = bad_word_ids.replace("['", "{").replace("']", "}").replace("'", "")
 
18
  bad_word = " " + bad_word
19
  ids = tokenizer(bad_word).input_ids
20
  ids = str(ids)
21
+ ids = ids.replace("]", ": -30").replace("[", "").replace(", ", ":-30, ")
22
  bad_word_ids.append(ids)
23
  bad_word_ids = str(bad_word_ids)
24
  bad_word_ids = bad_word_ids.replace("['", "{").replace("']", "}").replace("'", "")