Update app.py

app.py

@@ -329,6 +329,37 @@ def infer(
                         controlnet_conditioning_scale=controlnet_conditioning_scale,
                         generator=generator,
                     ).images[0]
+                else    
+
+                    if ip_adapter_mode == "face_model":     
+
+                        print('ip_adapter_mode = ', ip_adapter_mode)
+
+                        # Преобразование изображений для IP-Adapter (размер 224x224)
+                        ip_source_image = preprocess_image(ip_source_image, width, height, resize_to_224=True)
+                        ip_adapter_image = preprocess_image(ip_adapter_image, width, height, resize_to_224=True)
+                        
+                        pipe_ip_adapter = StableDiffusionPipeline.from_pretrained(
+                            model_default,
+                            torch_dtype=torch_dtype,
+                        ).to(device)
+
+                        pipe_ip_adapter.scheduler = DDIMScheduler.from_config(pipeline.scheduler.config)
+                        pipe_ip_adapter.load_ip_adapter(IP_ADAPTER, subfolder="models", weight_name=WEIGHT_NAME_face)
+
+                        generator = torch.Generator(device).manual_seed(seed)
+
+                        pipe_ip_adapter.set_ip_adapter_scale(ip_adapter_strength)
+                        image = pipe_ip_adapter(
+                            prompt=prompt,
+                            negative_prompt=negative_prompt,
+                            ip_adapter_image=ip_adapter_image,
+                            width=width,
+                            height=height,
+                            guidance_scale=guidance_scale,
+                            num_inference_steps=num_inference_steps,
+                            generator=generator,
+                        ).images[0]
     else:
         # Генерация изображений с ControlNet ----------------------------------------------------------------------------------------------------------------
 
@@ -512,11 +543,13 @@ examples = [
     "The smiling man. His face and hands are visible. Anime style. The best quality.",
     "The smiling girl. Anime style. Best quality, high quality.",
     "lego batman and robin. Rich and vibrant colors.",
+    "Photo of Pushkin as a hockey player in uniform with a stick, playing hockey on the ice arena in the NHL.",
 ]    
 
 examples_negative = [
     "Blurred details, low resolution, bad anatomy, no face visible, poor image of a man's face, poor quality, artifacts, black and white image.",
     "Monochrome, lowres, bad anatomy, worst quality, low quality",
+    "lowres, bad anatomy, worst quality, low quality, black and white image.",
 ]
 
 css = """
@@ -701,6 +734,7 @@ with gr.Blocks(css=css) as demo:
                         "pose_estimation",
                         "edge_detection", 
                         "depth_map",
+                        "face_model"
                     ],
                     value="pose_estimation",
                     interactive=True,