Update app.py

app.py

@@ -27,6 +27,7 @@ def infer(
     guidance_scale,
     lora_scale,
     num_inference_steps,
+    remove_background,
     progress=gr.Progress(track_tqdm=True),
 ):
     if randomize_seed:
@@ -86,6 +87,10 @@ with gr.Blocks(css=css) as demo:
                 info="Choose which diffusion model to use"
             )
 
+        with gr.Row():
+                remove_background = gr.Checkbox(
+                    label="Delete background?", value=True
+                )
             
         with gr.Row():
             prompt = gr.Text(
@@ -117,10 +122,6 @@ with gr.Blocks(css=css) as demo:
             )
 
             randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
-            with gr.Row():
-                remove_background = gr.Checkbox(
-                    label="Delete background?", value=True
-                )
                 
             with gr.Row():
                 width = gr.Slider(
@@ -178,7 +179,7 @@ with gr.Blocks(css=css) as demo:
             guidance_scale,
             lora_scale,
             num_inference_steps,
-            remove_backgroung
+            remove_background
         ],
         outputs=[result, seed],
     )