diff --git a/api/client.py b/api/client.py
index 873a195..4fd751c 100644
--- a/api/client.py
+++ b/api/client.py
@@ -11,7 +11,7 @@
 
 ENDPOINT = "http://127.0.0.1:8001"
 if "REMOTE_URL_RAILWAY" in os.environ:
-    URL = os.environ["REMOTE_URL_RAILWAY"]
+    ENDPOINT = os.environ["REMOTE_URL_RAILWAY"]
 
 print(f"API ENDPOINT: {ENDPOINT}")
 
diff --git a/requirements.txt b/requirements.txt
index d3e0960..7bbeed9 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,7 +2,7 @@ e2cnn
 einops
 easydict
 gdown
-gradio==5.1.0
+gradio==5.4.0
 h5py
 huggingface_hub
 imageio