Update app.py

app.py

@@ -105,6 +105,27 @@ def remove_background(image_ori):
     # Return images in the specified order
     return foreground, background, pred_pil, reverse_mask
 
+# Custom CSS for button styling
+custom_css = """
+@keyframes gradient-animation {
+    0% { background-position: 0% 50%; }
+    50% { background-position: 100% 50%; }
+    100% { background-position: 0% 50%; }
+}
+
+.gradio-container button {
+    background: linear-gradient(
+        135deg,
+        #e0f7fa, #e8f5e9, #fff9c4, #ffebee,
+        #f3e5f5, #e1f5fe, #fff3e0, #e8eaf6
+    );
+    background-size: 400% 400%;
+    animation: gradient-animation 15s ease infinite;
+    border-radius: 12px;
+    color: white;
+}
+"""
+
 iface = gr.Interface(
     fn=remove_background_wrapper,
     inputs=gr.Image(type="numpy"),
@@ -114,7 +135,8 @@ iface = gr.Interface(
         gr.Image(type="pil", label="Foreground Mask"),
         gr.Image(type="pil", label="Background Mask")
     ],
-    allow_flagging="never"
+    allow_flagging="never",
+    css=custom_css  # Apply custom CSS here
 )
 
 if __name__ == "__main__":