add catch for ignoring the token highlighting

demo_watermark.py

@@ -408,6 +408,11 @@ def detect(input_text, args, tokenizer, device=None, return_green_token_mask=Tru
         output = [["Error","string too short to compute metrics"]]
         output += [["",""] for _ in range(6)]
 
+    # for now, just don't display the green token mask 
+    # if we're using normalizers or ignore_repeated_bigrams
+    if args.normalizers != [] or args.ignore_repeated_bigrams:
+        green_token_mask = None
+
     html_output = ""
     if green_token_mask is not None:
         # hack bc we need a fast tokenizer with charspan support