Update app.py

app.py

@@ -60,8 +60,14 @@ def process_image(image, num_steps=10, guidance_scale=1.0):
     except Exception as e:
         return None, None, f"❌ Error: {str(e)}"
 
+css = """
+.image-container img {
+    object-fit: contain !important;
+}
+"""
+
 # Create Gradio interface
-with gr.Blocks(title="Fibo-Edit-RMBG - Background Removal") as demo:
+with gr.Blocks(title="Fibo-Edit-RMBG - Background Removal", css=css) as demo:
     gr.Markdown("""
     # Fibo-Edit-RMBG - Background Removal
 
@@ -106,7 +112,7 @@ with gr.Blocks(title="Fibo-Edit-RMBG - Background Removal") as demo:
             output_image = ImageSlider(
                 label="Result vs Original",
                 type="pil",
-                height=400
+                # height=400
             )
 
             download_file = gr.File(label="Download PNG (with transparency)")