Update app.py

app.py

@@ -14,34 +14,37 @@ def remove_background(image):
         raise TypeError("Unsupported image type")
     return output
 
-iface = gr.Interface(
-    fn=remove_background,
-    inputs=gr.Image(label="Upload Image"),
-    outputs=gr.Image(label="Output Image"),
-    layout="vertical",
-    css="""
-    .container { margin-top: 0px !important; }
+css = """
+    body {
+        background-color: #f5f5f5;
+        font-family: Arial, sans-serif;
+    }
     .input_image, .output_image {
-        display: flex;
-        justify-content: center;
-        align-items: center;
-        margin: 10px 0;
+        border-radius: 10px;
+        box-shadow: 0px 4px 10px rgba(0, 0, 0, 0.1);
     }
-    .button {
+    .gr-button {
         background-color: #4CAF50;
         color: white;
         border: none;
         padding: 10px 20px;
         text-align: center;
-        text-decoration: none;
-        display: inline-block;
         font-size: 16px;
-        margin: 4px 2px;
         cursor: pointer;
-        border-radius: 8px;
+        border-radius: 5px;
+        margin: 4px 2px;
     }
-    """
+    .gr-button:hover {
+        background-color: #45a049;
+    }
+"""
+
+iface = gr.Interface(
+    fn=remove_background,
+    inputs=gr.Image(label="Upload Image", css_class="input_image"),
+    outputs=gr.Image(label="Output Image", css_class="output_image"),
+    live=True,
+    css=css
 )
 
-if __name__ == "__main__":
-    iface.launch()
+iface.launch()