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,6 +98,12 @@ 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)
+        
         # Преобразуем изображения
         source_image = preprocess_image(source_image, width, height)
         control_image = preprocess_image(control_image, width, height)
@@ -176,10 +182,11 @@ def infer(
 
 examples = [
     "A young man in anime style. The image is characterized by high definition and resolution. Handsome, thoughtful man, attentive eyes. The man is depicted in the foreground, close-up or in the middle. High-quality images of the face, eyes, nose, lips, hands and clothes. The background and background are blurred and indistinct. The play of light and shadow is visible on the face and clothes.",
+    "A man runs through the park against the background of trees. The man's entire figure, face, arms and legs are visible. Anime style. The best quality.",
 ]    
 
 examples_negative = [
-    "blurred details, low resolution, poor image of a man's face, poor quality, artifacts, black and white image",
+    "Blurred details, low resolution, poor image of a man's face, poor quality, artifacts, black and white image",
 ]
 
 css = """
@@ -273,6 +280,7 @@ with gr.Blocks(css=css) as demo:
                     value=512,
                 )
 
+        # ControlNet ---------------------------------------------------------------------------------
         with gr.Blocks():
             with gr.Row():
                 use_control_net = gr.Checkbox(
@@ -306,6 +314,7 @@ with gr.Blocks(css=css) as demo:
                 inputs=use_control_net,
                 outputs=control_net_options,
             )
+        # --------------------------------------------------------------------------------------
 
         gr.Examples(examples=examples, inputs=[prompt])
         gr.Examples(examples=examples_negative, inputs=[negative_prompt])