Update app.py

app.py

@@ -66,6 +66,7 @@ def remove_background(image):
         raise gr.Error("Please upload an image.")
 
     image_ori = Image.fromarray(image).convert('RGB')
+    original_size = image_ori.size
     
     # Preprocess the image
     image_preprocessor = ImagePreprocessor(resolution=(1024, 1024))
@@ -79,8 +80,9 @@ def remove_background(image):
 
     # Show Results
     pred_pil = transforms.ToPILImage()(pred)
+    pred_pil = pred_pil.resize(original_size, Image.BICUBIC)  # Resize mask to original size
     image_masked = refine_foreground(image_ori, pred_pil)
-    image_masked.putalpha(pred_pil.resize(image_ori.size))
+    image_masked.putalpha(pred_pil)
 
     torch.cuda.empty_cache()
 
@@ -99,6 +101,13 @@ body {
     font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, Helvetica, Arial, sans-serif, "Apple Color Emoji", "Segoe UI Emoji", "Segoe UI Symbol";
 }
 .gradio-container {
+    background: white;
+}
+.gradio-button {
+    font-family: inherit;
+    font-size: 16px;
+    font-weight: bold;
+    color: #000000;
     background: linear-gradient(
         135deg,
         #e0f7fa, #e8f5e9, #fff9c4, #ffebee,
@@ -106,24 +115,23 @@ body {
     );
     background-size: 400% 400%;
     animation: gradient-animation 15s ease infinite;
+    border: 2px solid black;
+    border-radius: 10px;
+}
+.gradio-button:hover {
+    background: linear-gradient(
+        135deg,
+        #b2ebf2, #c8e6c9, #fff176, #ffcdd2,
+        #e1bee7, #b3e5fc, #ffe0b2, #c5cae9
+    );
+    background-size: 400% 400%;
+    animation: gradient-animation 15s ease infinite;
 }
 @keyframes gradient-animation {
     0% { background-position: 0% 50%; }
     50% { background-position: 100% 50%; }
     100% { background-position: 0% 50%; }
 }
-.gradio-button {
-    font-family: inherit;
-    font-size: 16px;
-    font-weight: bold;
-    color: #000000;
-    background: white;
-    border: 2px solid black;
-    border-radius: 10px;
-}
-.gradio-button:hover {
-    background: #f0f0f0;
-}
 """
 
 iface = gr.Interface(
@@ -133,8 +141,6 @@ iface = gr.Interface(
         gr.Image(type="filepath", label="Mask"),
         gr.Image(type="filepath", label="Output")
     ],
-    title="<div style='font-size: 36px; font-weight: bold;'>{.Remove Background}</div>",
-    description="Upload an image to remove its background using BiRefNet.",
     allow_flagging="never",
     css=css
 )