Update app.py

app.py

@@ -37,8 +37,8 @@ def find_path(name: str, path: str = None) -> str:
 def add_comfyui_directory_to_sys_path() -> None:
     comfyui_path = find_path("ComfyUI")
     if comfyui_path is not None and os.path.isdir(comfyui_path):
-        sys.path.append(comfyui_path)
-        print(f"'{comfyui_path}' added to sys.path")
+        sys.path.insert(0, comfyui_path)
+        print(f"'{comfyui_path}' inserted to sys.path")
 
 def add_extra_model_paths() -> None:
     try:
@@ -55,13 +55,13 @@ def add_extra_model_paths() -> None:
 def import_custom_nodes() -> None:
     import asyncio
     import execution
-    from nodes import init_extra_nodes
+    from nodes import init_custom_nodes
     import server
     loop = asyncio.new_event_loop()
     asyncio.set_event_loop(loop)
     server_instance = server.PromptServer(loop)
     execution.PromptQueue(server_instance)
-    init_extra_nodes()
+    init_custom_nodes()
 
 # Setup ComfyUI and custom nodes
 if not os.path.exists("ComfyUI"):
@@ -106,7 +106,7 @@ if not os.path.exists(esrgan_x4_path):
 add_comfyui_directory_to_sys_path()
 add_extra_model_paths()
 from folder_paths import add_model_folder_path
-add_model_folder_path("unet", "ComfyUI/models/diffusion_models")
+add_model_folder_path("unet", find_path("ComfyUI/models/diffusion_models"))
 import_custom_nodes()
 
 from nodes import NODE_CLASS_MAPPINGS