Update app.py

app.py

@@ -1,4 +1,10 @@
 import os
+
+# Set a default watermark key to avoid the NoneType error
+# Do this BEFORE any imports
+if "WATERMARK_KEY" not in os.environ:
+    os.environ["WATERMARK_KEY"] = "0 0 0 0"  # Default placeholder
+
 import subprocess
 import tempfile
 
@@ -10,10 +16,6 @@ import torchaudio
 from generator import Segment, load_csm_1b
 from watermarking import watermark
 
-# Set a default watermark key to avoid the NoneType error
-if "WATERMARK_KEY" not in os.environ:
-    os.environ["WATERMARK_KEY"] = "0 0 0 0"  # Default placeholder
-
 # Simplified environment variables handling
 gpu_timeout = int(os.getenv("GPU_TIMEOUT", 60))