fix HF Space deployment (#248)

mljar · Apr 21, 2023 · 122b0a8 · 122b0a8
1 parent be211f9
commit 122b0a8
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/mercury/apps/nbworker/__main__.py b/mercury/apps/nbworker/__main__.py
@@ -48,6 +48,9 @@
 server_url = sys.argv[4]
 
 
+if os.environ.get("MERCURY_SERVER_URL") is None:
+    os.environ["MERCURY_SERVER_URL"] = server_url.replace("ws://", "http://").replace("wss://", "https://")
+
 def signal_handler(signal, frame):
     global stop_event
     log.debug("\nBye bye!")