Update controlnet_facefix.py

controlnet_facefix.py

@@ -1,6 +1,6 @@
 # controlnet_facefix.py - EINFACHE VERSION (GANZES BILD VERBESSERN)
 import torch
-from diffusers import StableDiffusionControlNetInpaintPipeline, ControlNetModel
+from diffusers import StableDiffusionControlNetPipeline, ControlNetModel
 from PIL import Image
 import time
 import cv2
@@ -97,7 +97,7 @@ def apply_facefix(image: Image.Image, prompt: str, negative_prompt: str, seed: i
     if _pipeline is None:
         try:
             print("🔄 Lade Face-Fix Pipeline...")
-            _pipeline = StableDiffusionControlNetInpaintPipeline.from_pretrained(
+            _pipeline = StableDiffusionControlNetPipeline.from_pretrained(
                 model_id,
                 controlnet=[_controlnet_pose, _controlnet_depth],
                 torch_dtype=torch.float16,
@@ -130,17 +130,15 @@ def apply_facefix(image: Image.Image, prompt: str, negative_prompt: str, seed: i
         result = pipeline(
             prompt=enhanced_prompt,
             negative_prompt=enhanced_negative,
-            image=image.resize((512, 512)),
-            mask_image=None,  # ← WICHTIG: None = ganzes Bild!
-            control_image=[pose_img, depth_img],
-            controlnet_conditioning_scale=[0.7, 0.5],  # Mittel für subtile Verbesserung
-            strength=0.3,  # Niedrig für feine Anpassungen
+            image=[pose_img, depth_img],  # ← WICHTIG: control_image als Liste
+            controlnet_conditioning_scale=[0.7, 0.5],
             num_inference_steps=20,
             guidance_scale=7.0,
             generator=torch.Generator(device).manual_seed(seed),
             height=512,
             width=512,
         ).images[0]
+
         
         # Zurück auf Originalgröße
         if image.size != (512, 512):