Update demo_watermark.py

demo_watermark.py

@@ -501,7 +501,7 @@ def run_gradio(args, model=None, device=None, tokenizer=None):
         check_prompt_partial = partial(check_prompt, model=model, device=device)
         generate_partial = partial(generate, model=model, device=device)
         detect_partial = partial(detect, device=device)
-'''
+
         with gr.Tab("Welcome"):
             with gr.Row():
                 with gr.Column(scale=2):
@@ -545,7 +545,7 @@ def run_gradio(args, model=None, device=None, tokenizer=None):
                         ![](https://drive.google.com/uc?export=view&id=1yVLPcjm-xvaCjQyc3FGLsWIU84v1QRoC)
                         """
                     )
-                  '''  
+                  
 
         with gr.Tab("Generate & Detect"):
             
@@ -577,7 +577,7 @@ def run_gradio(args, model=None, device=None, tokenizer=None):
                     return redecoded_input + f"\n\n[Prompt was truncated before generation due to length...]", args
                 else: 
                     return orig_prompt, args
-        '''
+        
         with gr.Tab("Detector Only"):
             with gr.Row():
                 with gr.Column(scale=2):
@@ -589,7 +589,7 @@ def run_gradio(args, model=None, device=None, tokenizer=None):
                     detection_result = gr.Dataframe(headers=["Metric", "Value"], interactive=False,row_count=7,col_count=2)
             with gr.Row():
                     detect_btn = gr.Button("Detect")
-        '''
+        
         # Parameter selection group
         with gr.Accordion("Advanced Settings",open=False):
             with gr.Row():
@@ -703,7 +703,7 @@ def run_gradio(args, model=None, device=None, tokenizer=None):
                 the confidence of the detection based on the unlikeliness of this `z-score` observation.
             """
             )
-'''
+
         gr.HTML("""
                 <p>For faster inference without waiting in queue, you may duplicate the space and upgrade to GPU in settings. 
                     Follow the github link at the top and host the demo on your own GPU hardware to test out larger models.
@@ -712,7 +712,7 @@ def run_gradio(args, model=None, device=None, tokenizer=None):
                 <img style="margin-top: 0em; margin-bottom: 0em" src="https://bit.ly/3gLdBN6" alt="Duplicate Space"></a>
                 <p/>
                 """) 
-                '''
+                
         
         # Register main generation tab click, outputing generations as well as a the encoded+redecoded+potentially truncated prompt and flag, then call detection
         generate_btn.click(fn=check_prompt_partial, inputs=[prompt,session_args,session_tokenizer], outputs=[redecoded_input, truncation_warning, session_args]).success(