Spaces:

jianuo
/

LLMwatermark

Sleeping

jianuo commited on Aug 10, 2024

Commit

905259f

•

1 Parent(s): 6ee67ba

change a text

Files changed (1) hide show

demo_watermark.py CHANGED Viewed

@@ -490,7 +490,7 @@ def detect(input_text, args, tokenizer, device=None, return_green_token_mask=Tru
         if score_dict['prediction']:
             html_look = gr.HTML("""<div style="width: 100%; font-size: 24px; height: 100px; border-radius: 20px; background-color: rgba(255, 0, 0, 0.25); display: flex; justify-content: center; align-items: center; color: white; font-weight: bold;">
-        <span>有 “{}” 的水印</span>
     </div>""".format(name), visible=True)
         else:

         if score_dict['prediction']:
             html_look = gr.HTML("""<div style="width: 100%; font-size: 24px; height: 100px; border-radius: 20px; background-color: rgba(255, 0, 0, 0.25); display: flex; justify-content: center; align-items: center; color: white; font-weight: bold;">
+        <span>含有水印：“{}”</span>
     </div>""".format(name), visible=True)
         else: