Update app.py

app.py

@@ -9,7 +9,7 @@ from huggingface_hub import hf_hub_download
 import spaces
 from typing import Union, Sequence, Mapping, Any
 
-# Configuração inicial e diagnóstico CUDA
+# Diagnóstico CUDA
 print("Python version:", sys.version)
 print("Torch version:", torch.__version__)
 print("CUDA disponível:", torch.cuda.is_available())
@@ -23,9 +23,10 @@ comfyui_path = os.path.join(current_dir, "ComfyUI")
 sys.path.append(comfyui_path)
 
 # Importar ComfyUI components
-from nodes import NODE_CLASS_MAPPINGS, init_extra_nodes
-from comfy import model_management
+import execution
+from nodes import NODE_CLASS_MAPPINGS
 import folder_paths
+from comfy import model_management
 
 # Configuração de diretórios
 BASE_DIR = os.path.dirname(os.path.realpath(__file__))
@@ -33,9 +34,8 @@ output_dir = os.path.join(BASE_DIR, "output")
 os.makedirs(output_dir, exist_ok=True)
 folder_paths.set_output_directory(output_dir)
 
-# Inicializar nós extras
-print("Inicializando nós extras...")
-init_extra_nodes()
+# Inicializar execução
+execution.init()
 
 # Helper function
 def get_value_at_index(obj: Union[Sequence, Mapping], index: int) -> Any:
@@ -54,7 +54,7 @@ def download_models():
         ("black-forest-labs/FLUX.1-dev", "ae.safetensors", "models/vae"),
         ("black-forest-labs/FLUX.1-dev", "flux1-dev.safetensors", "models/diffusion_models"),  # Corrigido aqui
         ("google/siglip-so400m-patch14-384", "model.safetensors", "models/clip_vision"),
-        ("nftnik/NFTNIK-FLUX.1-dev-LoRA", "NFTNIK_FLUX.1[dev]_LoRA.safetensors", "models/lora")
+        ("nftnik/NFTNIK-FLUX.1-dev-LoRA", "NFTNIK_V1.safetensors", "models/lora")
     ]
     
     for repo_id, filename, local_dir in models: