Update app.py

app.py

@@ -126,24 +126,20 @@ custom_css = """
 }
 """
 
-iface = gr.Interface(
-    fn=remove_background_wrapper,
-    inputs=gr.Image(type="numpy"),
-    outputs=[
-        gr.Image(type="pil", label="Foreground"),
-        gr.Image(type="pil", label="Background"),
-        gr.Image(type="pil", label="Foreground Mask"),
-        gr.Image(type="pil", label="Background Mask")
-    ],
-    allow_flagging="never",
-    css=custom_css,
-    live=False
-)
-
-# Assigning an ID to the submit button
 with gr.Blocks(css=custom_css) as demo:
-    btn = gr.Button("Submit", elem_id="submit-button")
-    demo.launch()
-
-if __name__ == "__main__":
-    iface.launch(debug=True)
+    # Interface setup with input and output
+    with gr.Row():
+        with gr.Column():
+            image_input = gr.Image(type="numpy", label="Upload Image")
+            btn = gr.Button("Submit", elem_id="submit-button")
+        with gr.Column():
+            output_foreground = gr.Image(type="pil", label="Foreground")
+            output_background = gr.Image(type="pil", label="Background")
+            output_foreground_mask = gr.Image(type="pil", label="Foreground Mask")
+            output_background_mask = gr.Image(type="pil", label="Background Mask")
+
+    # Link the button to the processing function
+    btn.click(fn=remove_background_wrapper, inputs=image_input, outputs=[
+        output_foreground, output_background, output_foreground_mask, output_background_mask])
+
+    demo.launch(debug=True)