Update app.py

app.py

@@ -43,14 +43,16 @@ def generate(image: Image.Image, edit_prompt: str):
     
     global sampler
     if sampler is None:
-        # CORREÇÃO: Inicializa o sampler usando os argumentos do arquivo .yaml
-        print("Inicializando o XFluxSampler com a configuração...")
+        # A inicialização do sampler agora lê os novos parâmetros do .yaml
+        print("Inicializando o XFluxSampler com a configuração completa...")
         sampler = XFluxSampler(
             device=device,
             ip_loaded=args.use_ip,
             spatial_condition=args.use_spatial_condition,
             share_position_embedding=args.share_position_embedding,
-            use_share_weight_referencenet=args.use_share_weight_referencenet
+            use_share_weight_referencenet=args.use_share_weight_referencenet,
+            double_block_refnet=args.double_block_refnet,
+            single_block_refnet=args.single_block_refnet
         )
 
     img = image_resize(image, 544)
@@ -59,7 +61,6 @@ def generate(image: Image.Image, edit_prompt: str):
     img = torch.from_numpy((np.array(img) / 127.5) - 1)
     img = img.permute(2, 0, 1).unsqueeze(0).to(device, dtype=dtype)
 
-    # CORREÇÃO: Passa a imagem de origem se qualquer modo de condicionamento estiver ativo
     use_image_conditioning = args.use_spatial_condition or args.use_share_weight_referencenet
     
     result = sampler(