Update app.py

app.py

@@ -93,24 +93,15 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column(scale=1):
             input_image = gr.Image(type="numpy", label="Upload Image")
-            process_btn = gr.Button("Remove Background", variant="primary")
         
         with gr.Column(scale=1):
             output_image = gr.Image(type="numpy", label="Result")
-            download_btn = gr.File(label="Download Image")
+            download_btn = gr.File(label="Download Processed Image", elem_id="download_button")
     
     # Set up example images
     gr.Examples(examples, inputs=input_image)
     
-    # Set up processing logic
-    process_btn.click(
-        fn=process,
-        inputs=input_image,
-        outputs=[output_image, download_btn],
-        show_progress="full"
-    )
-    
-    # Also process automatically when image is uploaded
+    # Process automatically when image is uploaded
     input_image.change(
         fn=process,
         inputs=input_image,
@@ -118,5 +109,14 @@ with gr.Blocks() as demo:
         show_progress="full"
     )
 
+    # Style to ensure download button is visible
+    gr.HTML("""
+    <style>
+    #download_button {
+        margin-top: 10px;
+    }
+    </style>
+    """)
+
 if __name__ == "__main__":
     demo.launch(share=False)