Update app.py

app.py

@@ -9,6 +9,11 @@ import spaces
 from comfy import model_management
 from nodes import NODE_CLASS_MAPPINGS
 
+sys.path.append(os.path.abspath(os.path.dirname(__file__)))
+
+add_comfyui_directory_to_sys_path()
+add_extra_model_paths()
+
 hf_hub_download(
     repo_id="Comfy-Org/stable-diffusion-v1-5-archive",
     filename="v1-5-pruned-emaonly-fp16.safetensors",
@@ -96,10 +101,6 @@ def add_extra_model_paths() -> None:
         print("Could not find the extra_model_paths config file.")
 
 
-add_comfyui_directory_to_sys_path()
-add_extra_model_paths()
-
-
 def import_custom_nodes() -> None:
     """Find all custom nodes in the custom_nodes folder and add those node objects to NODE_CLASS_MAPPINGS
 
@@ -122,9 +123,6 @@ def import_custom_nodes() -> None:
     # Initializing custom nodes
     init_extra_nodes()
 
-
-from nodes import NODE_CLASS_MAPPINGS
-
 @spaces.GPU(duration=60)
 def generate_image(positive_prompt, negative_prompt):
     import_custom_nodes()