diff --git a/proxy-server/main.py b/proxy-server/main.py
index f44dd1025..71b5b6a95 100644
--- a/proxy-server/main.py
+++ b/proxy-server/main.py
@@ -61,7 +61,7 @@ def get_models():
 
 if __name__ == "__main__":
   from waitress import serve
-  serve(app, host="0.0.0.0", port=5000, threads=500)
+  serve(app, host="0.0.0.0", port=os.environ.get("PORT", 5000), threads=500)
 
 ############### Advanced ##########################