touch ups

install.py

@@ -207,8 +207,8 @@ def install_custom_nodes():
 
 
 if __name__ == "__main__":
-    # install_comfyui()
-    # install_custom_nodes()
-    # download_huggingface_models()
-    # download_and_extract_antelopev2()
+    install_comfyui()
+    install_custom_nodes()
+    download_huggingface_models()
+    download_and_extract_antelopev2()
     print("🎉 Setup Complete! Run `run.py` to start ComfyUI.")

run.py

@@ -1,16 +1,16 @@
 import os
 import subprocess
 
-BASE_PATH = "./ComfyUI"
+COMFYUI_PATH = "./ComfyUI"
 PORT = 8000
 
 def run_comfyui():
     """Launch ComfyUI with external access."""
-    os.chdir(BASE_PATH)
+    os.chdir(COMFYUI_PATH)
     print(f"🚀 Launching ComfyUI on port {PORT}...")
 
     subprocess.run(f"python main.py --listen 0.0.0.0 --port {PORT} --disable-auto-launch", shell=True)
 
 if __name__ == "__main__":
     run_comfyui()
-    print(f"Now run port-forwarding\nssh -L {PORT}:localhost:{PORT} root@[IP address] -p [RUNPOD_PORT] -i ~/.ssh/id_runpod")
+    print(f"Now run port-forwarding\nssh -L {PORT}:localhost:{PORT} root@[IP_ADDRESS] -p [RUNPOD_PORT] -i ~/.ssh/[PRIVATE_KEY_NAME]")