Update app.py

app.py

@@ -61,11 +61,11 @@ def align_embeddings(prompt_embeds, negative_prompt_embeds):
 
 pipe_default = get_lora_sd_pipeline(lora_dir='./lora_man_animestyle', base_model_name_or_path=model_default, dtype=torch_dtype).to(device)
 #pipe_controlnet = StableDiffusionControlNetPipeline.from_pretrained(
-# pipe_controlnet = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(    
-#     model_default,
-#     controlnet=controlnet,
-#     torch_dtype=torch_dtype
-# ).to(device)
+pipe_controlnet = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(    
+    model_default,
+    controlnet=controlnet,
+    torch_dtype=torch_dtype
+).to(device)
 
 def preprocess_image(image, target_width, target_height):
     """
@@ -98,11 +98,11 @@ def infer(
     generator = torch.Generator(device).manual_seed(seed)
     
     if use_control_net and control_image is not None and source_image is not None:
-        pipe_controlnet = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(    
-            model_default,
-            controlnet=controlnet,
-            torch_dtype=torch_dtype
-        ).to(device)
+        # pipe_controlnet = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(    
+        #     model_default,
+        #     controlnet=controlnet,
+        #     torch_dtype=torch_dtype
+        # ).to(device)
         
         # Преобразуем изображения
         source_image = preprocess_image(source_image, width, height)
@@ -145,6 +145,7 @@ def infer(
             negative_prompt_embeds=negative_prompt_embeds,
             image=source_image,     
             control_image=control_image,  
+            strength=0.8, # Коэфф. зашумления, чем больше, тем больше меняется результирующее изображение относитенльно исходного
             width=width,
             height=height,
             num_inference_steps=num_inference_steps,
@@ -186,7 +187,7 @@ examples = [
 ]    
 
 examples_negative = [
-    "Blurred details, low resolution, poor image of a man's face, poor quality, artifacts, black and white image",
+    "Blurred details, low resolution, no face visible, poor image of a man's face, poor quality, artifacts, black and white image.",
 ]
 
 css = """