Update app.py

app.py

@@ -22,7 +22,7 @@ MAX_IMAGE_SIZE = 1024
 model_repo_id = "CompVis/stable-diffusion-v1-4"
 pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype, safety_checker=None).to(device)
 
-# Попробуем подгрузить LoRA-модификации
+# Попробуем подгрузить LoRA-модификации (unet + text_encoder)
 try:
     pipe.unet = PeftModel.from_pretrained(pipe.unet, "./unet")
     pipe.text_encoder = PeftModel.from_pretrained(pipe.text_encoder, "./text_encoder")
@@ -49,10 +49,9 @@ def infer(
 ):
     """
     Функция генерации изображения.
-    Параметры use_controlnet, control_strength, controlnet_mode, controlnet_image,
-    use_ip_adapter, ip_adapter_scale, ip_adapter_image — это заглушки для демонстрации UI.
+    В реальном проекте здесь нужно добавить интеграцию с ControlNet и IP-adapter.
+    Сейчас они служат лишь демонстрацией UI (заглушка).
     """
-
     global model_repo_id, pipe
 
     # Если пользователь ввёл другую модель, пробуем её загрузить с нуля
@@ -62,7 +61,7 @@ def infer(
 
         try:
             new_pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch_dtype).to(device)
-            # Повторно подгружаем LoRA
+            # Повторно подгружаем LoRA для нового пайплайна
             try:
                 new_pipe.unet = PeftModel.from_pretrained(new_pipe.unet, "./unet")
                 new_pipe.text_encoder = PeftModel.from_pretrained(new_pipe.text_encoder, "./text_encoder")
@@ -79,10 +78,7 @@ def infer(
     # Создаём генератор случайных чисел для детерминированности
     generator = torch.Generator(device=device).manual_seed(seed)
 
-    # --- Здесь должна быть интеграция ControlNet, IP-adapter и т.д. ---
-    # Для демонстрации интерфейса просто вызываем pipe как обычно.
-    # ------------------------------------------------------------------
-
+    # Заглушка: пока что просто вызываем pipe без учёта ControlNet/IP-adapter
     try:
         image = pipe(
             prompt=prompt,
@@ -189,7 +185,7 @@ with gr.Blocks(css=css) as demo:
                     value=512,
                 )
 
-            # --- Дополнительные элементы для ControlNet ---
+            # Блоки ControlNet
             use_controlnet = gr.Checkbox(label="Use ControlNet", value=False)
             with gr.Group(visible=False) as controlnet_group:
                 control_strength = gr.Slider(
@@ -206,11 +202,9 @@ with gr.Blocks(css=css) as demo:
                 )
                 controlnet_image = gr.Image(
                     label="ControlNet Image",
-                    source="upload",
                     type="pil"
                 )
 
-            # Функция для управления видимостью группы ControlNet
             def update_controlnet_group(use_controlnet):
                 return {"visible": use_controlnet}
 
@@ -220,7 +214,7 @@ with gr.Blocks(css=css) as demo:
                 outputs=[controlnet_group]
             )
 
-            # --- Дополнительные элементы для IP-adapter ---
+            # Блоки IP-adapter
             use_ip_adapter = gr.Checkbox(label="Use IP-adapter", value=False)
             with gr.Group(visible=False) as ip_adapter_group:
                 ip_adapter_scale = gr.Slider(
@@ -232,11 +226,9 @@ with gr.Blocks(css=css) as demo:
                 )
                 ip_adapter_image = gr.Image(
                     label="IP-adapter Image",
-                    source="upload",
                     type="pil"
                 )
 
-            # Функция для управления видимостью группы IP-adapter
             def update_ip_adapter_group(use_ip_adapter):
                 return {"visible": use_ip_adapter}