Update app.py

app.py

@@ -1,12 +1,55 @@
 import gradio as gr
 import numpy as np
 import random
+import time
 
 import spaces
 from diffusers import DiffusionPipeline
 import torch
 from PIL import Image
 
+from gradio.themes.base import Base
+from gradio.themes.utils import colors, fonts, sizes
+from typing import Iterable
+
+class Seafoam(Base):
+    def __init__(
+        self,
+        *,
+        primary_hue: colors.Color | str = colors.emerald,
+        secondary_hue: colors.Color | str = colors.blue,
+        neutral_hue: colors.Color | str = colors.gray,
+        spacing_size: sizes.Size | str = sizes.spacing_md,
+        radius_size: sizes.Size | str = sizes.radius_md,
+        text_size: sizes.Size | str = sizes.text_lg,
+        font: fonts.Font
+        | str
+        | Iterable[fonts.Font | str] = (
+            fonts.GoogleFont("Quicksand"),
+            "ui-sans-serif",
+            "sans-serif",
+        ),
+        font_mono: fonts.Font
+        | str
+        | Iterable[fonts.Font | str] = (
+            fonts.GoogleFont("IBM Plex Mono"),
+            "ui-monospace",
+            "monospace",
+        ),
+    ):
+        super().__init__(
+            primary_hue=primary_hue,
+            secondary_hue=secondary_hue,
+            neutral_hue=neutral_hue,
+            spacing_size=spacing_size,
+            radius_size=radius_size,
+            text_size=text_size,
+            font=font,
+            font_mono=font_mono,
+        )
+
+seafoam = Seafoam()
+
 # Device and model setup
 device = "cuda" if torch.cuda.is_available() else "cpu"
 model_repo_id = "stabilityai/stable-diffusion-3.5-large-turbo"
@@ -119,7 +162,7 @@ h1 { text-align: center; }
 footer { visibility: hidden; }
 '''
 
-with gr.Blocks(theme="YTheme/Minecraft", css=css) as demo:
+with gr.Blocks(theme=seafoam, css=css) as demo:
     gr.Markdown("## T2I SD3.5")
 
     with gr.Row():
@@ -187,5 +230,4 @@ with gr.Blocks(theme="YTheme/Minecraft", css=css) as demo:
     )
 
 if __name__ == "__main__":
-    #demo.launch(ssr_mode=False)
-    demo.launch()
+    demo.launch(ssr_mode=False)