Update app.py

app.py

@@ -16,8 +16,18 @@ class WhiteTheme(Base):
         self,
         *,
         primary_hue: colors.Color | str = colors.orange,
-        font: fonts.Font | str = fonts.GoogleFont("Inter"),
-        font_mono: fonts.Font | str = fonts.GoogleFont("Inter")
+        font: fonts.Font | str | tuple[fonts.Font | str, ...] = (
+            fonts.GoogleFont("Inter"),
+            "ui-sans-serif",
+            "system-ui",
+            "sans-serif",
+        ),
+        font_mono: fonts.Font | str | tuple[fonts.Font | str, ...] = (
+            fonts.GoogleFont("Inter"),
+            "ui-monospace",
+            "system-ui",
+            "monospace",
+        )
     ):
         super().__init__(
             primary_hue=primary_hue,
@@ -34,24 +44,21 @@ class WhiteTheme(Base):
             block_border_color="white",
             panel_border_color="white",
             input_border_color="lightgray",
+            shadow_drop="none",
             button_primary_background_fill="*primary_500",
             button_primary_background_fill_hover="*primary_600",
             button_primary_text_color="white",
             button_secondary_background_fill="white",
             button_secondary_border_color="lightgray",
-            block_shadow="none",
-            button_shadow="none",
-            input_shadow="none",
             slider_color="*primary_500",
-            slider_track_color="lightgray",
+            slider_track_color="lightgray"
         )
 
-# Your existing setup code
 torch.set_float32_matmul_precision('high')
 torch.jit.script = lambda f: f
+
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-# Keep all your existing functions unchanged
 def refine_foreground(image, mask, r=90):
     if mask.size != image.size:
         mask = mask.resize(image.size)
@@ -149,12 +156,11 @@ custom_css = """
     animation: gradient-animation 15s ease infinite;
     border-radius: 12px;
     color: black;
-    font-family: 'Inter', sans-serif;
 }
 """
 
-# Create the interface with the custom theme
 with gr.Blocks(css=custom_css, theme=WhiteTheme()) as demo:
+    # Interface setup with input and output
     with gr.Row():
         with gr.Column():
             image_input = gr.Image(type="numpy", sources=['upload'], label="Upload Image")
@@ -165,6 +171,7 @@ with gr.Blocks(css=custom_css, theme=WhiteTheme()) as demo:
             output_foreground_mask = gr.Image(type="pil", label="Foreground Mask")
             output_background_mask = gr.Image(type="pil", label="Background Mask")
 
+    # Link the button to the processing function
     btn.click(fn=remove_background_wrapper, inputs=image_input, outputs=[
         output_foreground, output_background, output_foreground_mask, output_background_mask])