Update app.py

app.py

@@ -13,6 +13,20 @@ import lpips
 from facenet_pytorch import MTCNN
 import legacy
 
+# --- THEME DEFINITION ---
+# This theme uses the colors and fonts from your website's CSS.
+theme = gr.themes.Base(
+    primary_hue=gr.themes.colors.blue, # A base blue hue
+    font=[gr.themes.GoogleFont("Poppins"), "sans-serif"],
+).set(
+    button_primary_background_fill="#4A90E2",      # Your --primary-color
+    button_primary_background_fill_hover="#357ABD", # Your --secondary-color
+    slider_color="#4A90E2",                        # Your --slider-thumb color
+    slider_color_dark="#4A90E2"
+)
+# -------------------------
+
+
 # --- Load All Models ---
 print("Loading all models...")
 device = torch.device("cpu")
@@ -84,7 +98,7 @@ def edit_uploaded_face(uploaded_image, strength, progress=gr.Progress(track_tqdm
     print("Processing complete.")
     return img_out
 
-# Use .queue() and the standard .launch(). This enables the API.
+# Create the Gradio Interface with the custom theme
 gr.Interface(
     fn=edit_uploaded_face,
     inputs=[
@@ -92,7 +106,7 @@ gr.Interface(
         gr.Slider(-5, 5, step=0.1, value=0, label="Gender Strength (← Feminine | Masculine →)")
     ],
     outputs=gr.Image(label="Edited Face"),
+    theme=theme, # <--- APPLYING YOUR CUSTOM THEME
     title="Face Editor Backend",
-    description="This engine detects a face in the uploaded image, then edits its gender expression. It is ready to be used as an API.",
-    flagging_options=None,
+    description="This engine detects a face in the uploaded image, then edits its gender expression.",
 ).queue().launch()