Update app.py

app.py

@@ -66,15 +66,22 @@ def process_image(image, num_steps=10, guidance_scale=1.0):
 # }
 # """
 
+# css = """
+# .image-container img,
+# .image-container .fixed img,
+# .image-container .fixed {
+#     object-fit: contain !important;
+#     max-height: 400px !important;
+# }
+# """
+
 css = """
-.image-container img,
-.image-container .fixed img,
-.image-container .fixed {
+#output-slider img {
     object-fit: contain !important;
-    max-height: 400px !important;
 }
 """
 
+
 # Create Gradio interface
 with gr.Blocks(title="Fibo-Edit-RMBG - Background Removal", css=css) as demo:
     gr.Markdown("""
@@ -121,7 +128,8 @@ with gr.Blocks(title="Fibo-Edit-RMBG - Background Removal", css=css) as demo:
             output_image = ImageSlider(
                 label="Result vs Original",
                 type="pil",
-                height=400
+                height=400,
+                elem_id="output-slider"
             )
 
             download_file = gr.File(label="Download PNG (with transparency)")