Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+print(f"Gradio version: {gr.__version__}")
 import numpy as np
 import random
 import spaces
@@ -6,7 +7,7 @@ import torch
 from diffusers import DiffusionPipeline, FlowMatchEulerDiscreteScheduler, AutoencoderTiny, AutoencoderKL
 from transformers import CLIPTextModel, CLIPTokenizer, T5EncoderModel, T5TokenizerFast
 from live_preview_helpers import calculate_shift, retrieve_timesteps, flux_pipe_call_that_returns_an_iterable_of_images
-import os  # Import os module to access environment variables
+import os
 
 # Retrieve the token from the environment variable
 hf_token = os.environ.get("HF_API_TOKEN")
@@ -73,52 +74,43 @@ css = """
     margin: 0 auto;
     max-width: 520px;
 }
-
 /* Optional: Additional dark mode customizations */
 body {
     background-color: #1e1e1e;
     color: #ffffff;
 }
-
 .gradio-container {
     background-color: #2c2c2c;
 }
-
 .gr-button, .gr-slider, .gr-checkbox {
     background-color: #3a3a3a;
     color: #ffffff;
 }
-
 .markdown {
     color: #ffffff;
 }
-
 .gr-accordion {
     background-color: #3a3a3a;
     color: #ffffff;
 }
-
 .gr-input, .gr-textbox, .gr-slider {
     background-color: #3a3a3a;
     color: #ffffff;
 }
-
 .gr-slider .gr-slider-track {
     background-color: #555555;
 }
-
 .gr-slider .gr-slider-thumb {
     background-color: #ffffff;
 }
-
 .gr-button:hover {
     background-color: #555555;
     color: #ffffff;
 }
 """
 
-# Define the dark theme using gr.themes.Dark()
-dark_theme = gr.themes.Dark()
+# Define the dark theme using gr.themes.Default with base='dark'
+dark_theme = gr.themes.Default(base="dark")
 
 with gr.Blocks(theme=dark_theme, css=css) as demo: