Update runcomfy.py

runcomfy.py

@@ -239,13 +239,14 @@ def main():
     comfyui_args = sys.argv[1:]
 
     # --- 필수 경로 설정 및 이동 ---
-    comfyui_dir = os.environ.get('WS', '/content') + '/ComfyUI'
+    comfyui_dir = WS + '/ComfyUI' 
+    # 또는 f-string 사용: comfyui_dir = f"{WS}/ComfyUI"
+    
     print(f"ComfyUI 디렉토리로 이동 중: {comfyui_dir}")
     try:
         os.chdir(comfyui_dir)
     except FileNotFoundError:
         print(f"오류: 디렉토리가 없습니다: {comfyui_dir}. 경로를 확인해 주세요.")
-        sys.exit(1)
 
     # 터널 스크립트 파일을 저장
     print("터널 스크립트 저장 중...")