pop prompt before state created

demo_watermark.py

@@ -351,8 +351,8 @@ def run_gradio(args, model=None, device=None, tokenizer=None):
                 )
 
         # Construct state for parameters, define updates and toggles
-        session_args = gr.State(value=args)
         default_prompt = args.__dict__.pop("default_prompt")
+        session_args = gr.State(value=args)
 
         with gr.Tab("Generate and Detect"):