Update app.py

app.py

@@ -83,13 +83,16 @@ def response(history, image):
         history[-1][1] = partial_response
         yield history
 
-with gr.Blocks(css=css, js="bg-remove2.js") as demo:
-    with gr.Column(elem_id="col-container"):
-        gr.HTML(title)
-        image = gr.Image(type="pil", elem_id="upload")
-        message = gr.Textbox(interactive=True, show_label=False, container=False)
-        chat = gr.Chatbot(show_label=False)
-        submit = gr.Button(value="Upload", variant="primary")
+with gr.Blocks(css=css) as demo:
+    gr.HTML(title)
+    with gr.Row()
+        with gr.Column(elem_id="col-container"):
+            image = gr.Image(type="pil")
+            message = gr.Textbox(interactive=True, show_label=False, container=False)
+            chat = gr.Chatbot(show_label=False)
+            submit = gr.Button(value="Upload", variant="primary")
+        with gr.Column()
+            output = gr.Image(type="pil")
         
     response_handler = (
         response,