Update app.py

app.py

@@ -18,8 +18,12 @@ load_dotenv()
 app = Flask(__name__)
 
 # Set server and websocket addresses from environment variables
-server_address = "https://gosign-de-comfyui.hf.space"
-ws_address = "wss://gosign-de-comfyui.hf.space/ws"
+# server_address = "https://gosign-de-comfyui.hf.space"
+# ws_address = "wss://gosign-de-comfyui.hf.space/ws"
+
+# Set server and websocket addresses from environment variables
+server_address = os.getenv("SERVER_ADDRESS")
+ws_address = os.getenv("WS_ADDRESS")
 
 # Example print statements to verify the addresses are loaded
 print(f"Server Address: {server_address}")