Update app.py

app.py

@@ -51,11 +51,17 @@ def run_model(input_image):
     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,
@@ -73,7 +79,11 @@ 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)
@@ -87,12 +97,20 @@ 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
 
@@ -193,12 +211,24 @@ class CustomTheme(gr.themes.Base):
         )
 
 css = """
-html, body, #root, #__next, .gradio-container, .gr-app {
+*, *::before, *::after {
+    box-sizing: border-box;
+}
+
+html, body {
     background-color: #191a1e !important;
     margin: 0 !important;
     padding: 0 !important;
+    height: 100vh;
+    width: 100vw;
+    overflow: hidden;
+}
+
+#root, #__next, .gradio-container, .gr-app {
+    background-color: #191a1e !important;
     height: 100%;
     width: 100%;
+    overflow: hidden;
 }
 
 .gr-block, .gr-row, .gr-column, .gr-box, .gr-examples, .gr-examples .gr-box, .gr-examples .gr-card {