Update app.py

app.py

@@ -43,7 +43,6 @@ example_files = [
     os.path.join("demo_files/examples", f) for f in os.listdir("demo_files/examples")
 ]
 
-# Define functions
 def run_model(input_image):
     start = time.time()
     with torch.no_grad():
@@ -129,7 +128,6 @@ def resize_foreground(
 
     # Compute padding according to the ratio
     new_size = int(new_image.shape[0] / ratio)
-    # Pad to new size
     ph0, pw0 = (new_size - size) // 2, (new_size - size) // 2
     ph1, pw1 = new_size - size - ph0, new_size - size - pw0
     new_image = np.pad(
@@ -144,7 +142,6 @@ def resize_foreground(
     return new_image
 
 def square_crop(input_image: Image) -> Image:
-    # Perform a center square crop
     min_size = min(input_image.size)
     left = (input_image.size[0] - min_size) // 2
     top = (input_image.size[1] - min_size) // 2
@@ -252,15 +249,21 @@ body {
     padding: 0;
 }
 
-/* Применяем фоновый цвет для контейнера Gradio */
+:root {
+    --background-fill-primary: #191a1e !important;
+    --background-fill-secondary: #191a1e !important;
+    --background-fill-tertiary: #191a1e !important;
+}
+
+/* Apply background to gradio container */
 .gradio-container {
     background-color: #191a1e !important;
 }
 
-/* Применяем фоновый цвет для блоков */
+/* Blocks background */
 .gr-block {
     background-color: #191a1e !important;
-    border: 1px solid #5271FF !important;
+    border: none !important;
 }
 
 /* Hide the footer */
@@ -292,6 +295,17 @@ h1, h2, h3, h4, h5, h6 {
     border: none;
     font-weight: bold;
 }
+
+/* Examples container */
+.gr-examples, .gr-examples .gr-box, .gr-examples .gr-card {
+    background-color: #191a1e !important;
+    border: none !important;
+}
+
+/* 3D Model container override if needed */
+.litmodel3d { 
+    background-color: #191a1e !important;
+}
 """
 
 # Build the Gradio interface
@@ -299,7 +313,7 @@ with gr.Blocks(theme=CustomTheme(), css=css) as demo:
     img_proc_state = gr.State()
     background_remove_state = gr.State()
 
-    with gr.Row(variant="panel"):
+    with gr.Row():  # removed variant="panel"
         with gr.Column():
             with gr.Row():
                 input_img = gr.Image(
@@ -409,5 +423,4 @@ with gr.Blocks(theme=CustomTheme(), css=css) as demo:
         ],
     )
 
-# Launch the interface
 demo.launch()