Update app.py

app.py

@@ -28,7 +28,6 @@ intrinsic, intrinsic_normed_cond = sf3d_utils.create_intrinsic_from_fov_deg(
     COND_FOVY_DEG, COND_HEIGHT, COND_WIDTH
 )
 
-
 model = SF3D.from_pretrained(
     "stabilityai/stable-fast-3d",
     config_name="config.yaml",
@@ -117,9 +116,9 @@ def resize_foreground(
         alpha[1].min(),
         alpha[1].max(),
     )
-    # crop the foreground
+    # Crop the foreground
     fg = image[y1:y2, x1:x2]
-    # pad to square
+    # Pad to square
     size = max(fg.shape[0], fg.shape[1])
     ph0, pw0 = (size - fg.shape[0]) // 2, (size - fg.shape[1]) // 2
     ph1, pw1 = size - fg.shape[0] - ph0, size - fg.shape[1] - pw0
@@ -130,9 +129,9 @@ def resize_foreground(
         constant_values=((0, 0), (0, 0), (0, 0)),
     )
 
-    # compute padding according to the ratio
+    # Compute padding according to the ratio
     new_size = int(new_image.shape[0] / ratio)
-    # pad to size, double side
+    # Pad to size, double side
     ph0, pw0 = (new_size - size) // 2, (new_size - size) // 2
     ph1, pw1 = new_size - size - ph0, new_size - size - pw0
     new_image = np.pad(
@@ -236,54 +235,56 @@ def update_foreground_ratio(img_proc, fr):
         foreground_res,
         gr.update(value=show_mask_img(foreground_res)),
     )
+
+# Corrected CustomTheme class
 class CustomTheme(gr.themes.Base):
-    def __init__(self):
-        super().__init__()
-        self.primary_hue = "#191a1e"
-        self.background_fill_primary = "#191a1e"
-        self.background_fill_secondary = "#191a1e"
-        self.background_fill_tertiary = "#191a1e"
-        self.text_color_primary = "#FFFFFF"
-        self.text_color_secondary = "#FFFFFF"
-        self.text_color_tertiary = "#FFFFFF"
-        self.input_background_fill = "#191a1e"
-        self.input_text_color = "#FFFFFF"
+    primary_hue = gr.themes.Color("#191a1e")
+    background_fill_primary = gr.themes.Color("#191a1e")
+    background_fill_secondary = gr.themes.Color("#191a1e")
+    background_fill_tertiary = gr.themes.Color("#191a1e")
+    text_color_primary = gr.themes.Color("#FFFFFF")
+    text_color_secondary = gr.themes.Color("#FFFFFF")
+    text_color_tertiary = gr.themes.Color("#FFFFFF")
+    input_background_fill = gr.themes.Color("#191a1e")
+    input_text_color = gr.themes.Color("#FFFFFF")
+
 css = """
-/* Задаем фоновый цвет всей страницы */
+/* Apply background color to the entire page */
 body {
-    background-color: #191a1e !important; /* Тёмный фон */
+    background-color: #191a1e !important;
     margin: 0;
     padding: 0;
 }
 
-/* Применяем шрифты */
+/* Apply fonts */
 @import url('https://fonts.googleapis.com/css2?family=Poppins:wght@400;500;700&display=swap');
 body, input, button, textarea, select, .gr-button {
     font-family: 'Poppins', sans-serif;
     color: #FFFFFF;
 }
 
-/* Настройки заголовков */
+/* Header settings */
 h1, h2, h3, h4, h5, h6 {
     font-family: 'Poppins', sans-serif;
     font-weight: 700;
     color: #FFFFFF;
 }
 
-/* Цвет кнопок */
-.generate-button {
-    background-color: #5271FF !important; /* Голубой */
+/* Button colors */
+.gr-button {
+    background-color: #5271FF !important;
     color: #FFFFFF !important;
     border: none;
     font-weight: bold;
 }
 
-/* Цвет контейнеров */
+/* Container colors */
 .gr-block {
-    background-color: #1c1c24 !important; /* Голубоватый тёмный фон */
-    border: 1px solid #5271FF !important; /* Голубая рамка */
+    background-color: #1c1c24 !important;
+    border: 1px solid #5271FF !important;
 }
 """
+
 with gr.Blocks(theme=CustomTheme(), css=css) as demo:
     img_proc_state = gr.State()
     background_remove_state = gr.State()
@@ -292,7 +293,7 @@ with gr.Blocks(theme=CustomTheme(), css=css) as demo:
         with gr.Column():
             with gr.Row():
                 input_img = gr.Image(
-                    type="pil", label="Input Image", sources="upload", image_mode="RGBA"
+                    type="pil", label="Input Image", source="upload", image_mode="RGBA"
                 )
                 preview_removal = gr.Image(
                     label="Preview Background Removal",
@@ -330,8 +331,8 @@ with gr.Blocks(theme=CustomTheme(), css=css) as demo:
             with gr.Column(visible=False, scale=1.0) as hdr_row:
                 gr.Markdown("""## HDR Environment Map
 
-                Select an HDR environment map to light the 3D model. You can also upload your own HDR environment maps.
-                """)
+Select an HDR environment map to light the 3D model. You can also upload your own HDR environment maps.
+""")
 
                 with gr.Row():
                     hdr_illumination_file = gr.File(