nl-to-tag-jp-test

Runtime error

p1atdev commited on Feb 15

Commit

26ea8db

verified ·

1 Parent(s): 8617234

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -36,7 +36,7 @@ DEVICE = (
     torch.device("mps") if torch.backends.mps.is_available() else torch.device("cuda")
 )
-BAD_WORD_KEYWORDS = ["(medium)"]
 def get_bad_words_ids(tokenizer: PreTrainedTokenizerFast):

     torch.device("mps") if torch.backends.mps.is_available() else torch.device("cuda")
 )
+BAD_WORD_KEYWORDS = ["(medium)", " text", "(style)"]
 def get_bad_words_ids(tokenizer: PreTrainedTokenizerFast):