Update app.py

app.py

@@ -6,11 +6,18 @@ import torch
 import gradio as gr
 from huggingface_hub import hf_hub_download
 import spaces
-from comfy import model_management
-from nodes import NODE_CLASS_MAPPINGS
 from PIL import Image
 from folder_paths import folder_paths
 
+# Append ComfyUI path to sys.path
+current_dir = os.path.dirname(os.path.abspath(__file__))
+comfyui_path = os.path.join(current_dir, "ComfyUI")
+sys.path.append(comfyui_path)
+
+# Import Comfy modules
+from comfy import model_management
+from nodes import NODE_CLASS_MAPPINGS
+
 # Helper function to get values from objects
 def get_value_at_index(obj: Union[Sequence, Mapping], index: int) -> Any:
     try: