del_back

app.py

@@ -18,6 +18,7 @@ MAX_IMAGE_SIZE = 1024
 def infer(
         model_id,
         lora,
+        del_back,
         prompt,
         negative_prompt,
         seed,
@@ -98,6 +99,8 @@ with gr.Blocks(css=css) as demo:
             inputs=model_id, 
             outputs=lora)
         
+        with gr.Row():
+            del_back = gr.Checkbox(label="Delete background", value=False)
 
         with gr.Row():
             prompt = gr.Text(
@@ -175,6 +178,7 @@ with gr.Blocks(css=css) as demo:
         inputs=[
             model_id,
             lora,
+            del_back,
             prompt,
             negative_prompt,
             seed,