Update app.py

app.py

@@ -1,11 +1,8 @@
-!pip install controlnet_aux
-
 import gradio as gr
 import numpy as np
 import torch
 from diffusers import StableDiffusionPipeline, ControlNetModel, StableDiffusionControlNetPipeline
 from peft import PeftModel, LoraConfig
-from controlnet_aux import OpenposeDetector
 import os
 
 MAX_SEED = np.iinfo(np.int32).max
@@ -15,9 +12,8 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model_default = "stable-diffusion-v1-5/stable-diffusion-v1-5"
 torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
 
-# Инициализация ControlNet и OpenposeDetector
+# Инициализация ControlNet
 controlnet = ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-openpose", torch_dtype=torch_dtype)
-openpose = OpenposeDetector.from_pretrained("lllyasviel/ControlNet")
 
 def get_lora_sd_pipeline(
     lora_dir='./lora_man_animestyle',
@@ -81,18 +77,17 @@ def infer(
     lora_scale=0.5,
     use_control_net=False,  # Параметр для включения ControlNet
     control_strength=0.5,   # Сила влияния ControlNet
-    control_image=None,     # Контрольное изображение
+    control_image=None,     # Контрольное изображение (карта позы)
     progress=gr.Progress(track_tqdm=True)
 ):
     generator = torch.Generator(device).manual_seed(seed)
     
     if use_control_net and control_image is not None:
         # Используем ControlNet
-        pose_image = openpose(control_image)
         image = pipe_controlnet(
             prompt=prompt,
             negative_prompt=negative_prompt,
-            image=pose_image,
+            image=control_image,  # Используем загруженное изображение как карту позы
             width=width,
             height=height,
             num_inference_steps=num_inference_steps,
@@ -251,7 +246,7 @@ with gr.Blocks(css=css) as demo:
                     value="pose_estimation",
                 )
         
-                control_image = gr.Image(label="Upload Control Image")
+                control_image = gr.Image(label="Upload Control Image (Pose Map)")
 
             use_control_net.change(
                 fn=lambda x: gr.Row.update(visible=x),
@@ -287,3 +282,4 @@ with gr.Blocks(css=css) as demo:
 
 if __name__ == "__main__":
     demo.launch()
+