Update app.py

app.py

@@ -126,6 +126,7 @@ def infer(
     controlnet_conditioning_scale=0.5, # Сила влияния ControlNet
     ip_source_image=None,   # Исходное изображение IP_adapter
     ip_adapter_image=None,  # Контрольное изображение IP_adapter
+    remove_bg,              # Удаление фона с изображения
     progress=gr.Progress(track_tqdm=True)
 ):
 
@@ -534,6 +535,10 @@ def infer(
             }
 
             image = pipe(**params).images[0]
+
+    # Если выбрано удаление фона
+    if remove_bg:
+        image = remove(image)            
     
     return image 
     # ---------------------------------------------------------------------------------------------------------------------------------------------
@@ -751,6 +756,14 @@ with gr.Blocks(css=css) as demo:
                 outputs=ip_adapter_options
             )
         # ---------------------------------------------------------------------------------------------------------
+
+        # Checkbox для удаления фона
+        remove_bg = gr.Checkbox(
+            label="Remove Background",
+            value=False,
+            interactive=True
+        )
+        # ---------------------------------------------------------------------------------------------------------
         
         gr.Examples(examples=examples, inputs=[prompt], label="Examples for prompt:")
         gr.Examples(examples=examples_negative, inputs=[negative_prompt], label="Examples for negative prompt:")
@@ -784,6 +797,7 @@ with gr.Blocks(css=css) as demo:
             controlnet_conditioning_scale, # Сила влияния ControlNet
             ip_source_image,    # Исходное изображение IP_adapter
             ip_adapter_image,   # Контрольное изображение IP_adapter
+            remove_bg,          # Удаление фона с изображения
         ],
         outputs=[result],
     )