Update app.py

app.py

@@ -14,23 +14,19 @@ from PIL import Image
 import sf3d.utils as sf3d_utils
 from sf3d.system import SF3D
 
-# Initialize the rembg session
 rembg_session = rembg.new_session()
 
-# Constants
 COND_WIDTH = 512
 COND_HEIGHT = 512
 COND_DISTANCE = 1.6
 COND_FOVY_DEG = 40
 BACKGROUND_COLOR = [0.5, 0.5, 0.5]
 
-# Cached camera parameters
 c2w_cond = sf3d_utils.default_cond_c2w(COND_DISTANCE)
 intrinsic, intrinsic_normed_cond = sf3d_utils.create_intrinsic_from_fov_deg(
     COND_FOVY_DEG, COND_HEIGHT, COND_WIDTH
 )
 
-# Load the model
 model = SF3D.from_pretrained(
     "stabilityai/stable-fast-3d",
     config_name="config.yaml",
@@ -38,9 +34,7 @@ model = SF3D.from_pretrained(
 )
 model.eval().cuda()
 
-example_files = [
-    os.path.join("demo_files/examples", f) for f in os.listdir("demo_files/examples")
-]
+example_files = [os.path.join("demo_files/examples", f) for f in os.listdir("demo_files/examples")]
 
 def run_model(input_image):
     start = time.time()
@@ -53,23 +47,15 @@ def run_model(input_image):
 
     tmp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".glb")
     trimesh_mesh.export(tmp_file.name, file_type="glb", include_normals=True)
-
     print("Generation took:", time.time() - start, "s")
     return tmp_file.name
 
 def create_batch(input_image: Image) -> dict[str, Any]:
-    img_cond = (
-        torch.from_numpy(
-            np.asarray(input_image.resize((COND_WIDTH, COND_HEIGHT))).astype(np.float32)
-            / 255.0
-        )
-        .float()
-        .clip(0, 1)
-    )
+    img_cond = (torch.from_numpy(
+        np.asarray(input_image.resize((COND_WIDTH, COND_HEIGHT))).astype(np.float32) / 255.0
+    ).float().clip(0, 1))
     mask_cond = img_cond[:, :, -1:]
-    rgb_cond = torch.lerp(
-        torch.tensor(BACKGROUND_COLOR)[None, None, :], img_cond[:, :, :3], mask_cond
-    )
+    rgb_cond = torch.lerp(torch.tensor(BACKGROUND_COLOR)[None, None, :], img_cond[:, :, :3], mask_cond)
 
     batch_elem = {
         "rgb_cond": rgb_cond,
@@ -87,11 +73,7 @@ def checkerboard(squares: int, size: int, min_value: float = 0.5):
     base[1::2, ::2] = 1
     base[::2, 1::2] = 1
     repeat_mult = size // squares
-    return (
-        base.repeat(repeat_mult, axis=0)
-        .repeat(repeat_mult, axis=1)[:, :, None]
-        .repeat(3, axis=-1)
-    )
+    return (base.repeat(repeat_mult, axis=0).repeat(repeat_mult, axis=1)[:, :, None].repeat(3, axis=-1))
 
 def remove_background(input_image: Image) -> Image:
     return rembg.remove(input_image, session=rembg_session)
@@ -105,22 +87,12 @@ def resize_foreground(image: Image, ratio: float) -> Image:
     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
-    new_image = np.pad(
-        fg,
-        ((ph0, ph1), (pw0, pw1), (0, 0)),
-        mode="constant",
-        constant_values=0,
-    )
+    new_image = np.pad(fg, ((ph0, ph1), (pw0, pw1), (0, 0)), mode="constant", constant_values=0)
 
     new_size = int(new_image.shape[0] / ratio)
     ph0, pw0 = (new_size - size) // 2, (new_size - size) // 2
     ph1, pw1 = new_size - size - ph0, new_size - size - pw0
-    new_image = np.pad(
-        new_image,
-        ((ph0, ph1), (pw0, pw1), (0, 0)),
-        mode="constant",
-        constant_values=0,
-    )
+    new_image = np.pad(new_image, ((ph0, ph1), (pw0, pw1), (0, 0)), mode="constant", constant_values=0)
     new_image = Image.fromarray(new_image, mode="RGBA").resize((COND_WIDTH, COND_HEIGHT))
     return new_image
 
@@ -221,78 +193,47 @@ class CustomTheme(gr.themes.Base):
         )
 
 css = """
-html, body {
-    margin: 0;
-    padding: 0;
+html, body, #root, #__next, .gradio-container, .gr-app {
     background-color: #191a1e !important;
-    height: 100vh;
-    width: 100vw;
-}
-
-:root {
-    --background-fill-primary: #191a1e !important;
-    --background-fill-secondary: #191a1e !important;
-    --background-fill-tertiary: #191a1e !important;
-}
-
-.gradio-container, .gr-app {
-    background-color: #191a1e !important;
-    color: #FFFFFF !important;
+    margin: 0 !important;
+    padding: 0 !important;
     height: 100%;
     width: 100%;
 }
 
-/* Set blocks */
-.gr-block {
+body > div {
     background-color: #191a1e !important;
-    border: none !important;
-}
-
-/* Hide the footer */
-footer {
-    visibility: hidden;
-    height: 0;
-    margin: 0;
-    padding: 0;
-    overflow: hidden;
 }
 
-body, input, button, textarea, select, .gr-button {
-    font-family: 'Poppins', sans-serif;
-    color: #FFFFFF;
-}
-
-/* Header styles */
-h1, h2, h3, h4, h5, h6 {
-    font-family: 'Poppins', sans-serif;
-    font-weight: 700;
-    color: #FFFFFF;
+.gr-block, .gr-row, .gr-column, .gr-box, .gr-examples, .gr-examples .gr-box, .gr-examples .gr-card {
+    background-color: #191a1e !important;
+    border: none !important;
 }
 
-/* Increase button size */
 .generate-button {
     background-color: #5271FF !important;
     color: #FFFFFF !important;
     border: none;
     font-weight: bold;
-    font-size: 1.2em; /* Увел��чиваем размер шрифта */
-    padding: 0.75em 2em; /* Увеличиваем отступы внутри кнопки */
-    border-radius: 0.3em; /* Небольшое скругление */
-}
-
-/* Examples container */
-.gr-examples, .gr-examples .gr-box, .gr-examples .gr-card {
-    background-color: #191a1e !important;
-    border: none !important;
+    font-size: 1.2em;
+    padding: 0.75em 2em;
+    border-radius: 0.3em;
 }
 
-/* 3D Model container if needed */
-.litmodel3d { 
-    background-color: #191a1e !important;
+footer {
+    visibility: hidden;
+    height: 0;
+    margin: 0;
+    padding: 0;
+    overflow: hidden;
 }
 """
 
-with gr.Blocks(theme=CustomTheme(), css=css) as demo:
+with gr.Blocks(
+    theme=CustomTheme(), 
+    css=css,
+    style={"width":"100vw","height":"100vh","background":"#191a1e","margin":"0","padding":"0"}
+) as demo:
     img_proc_state = gr.State()
     background_remove_state = gr.State()