Update app.py

app.py

@@ -67,7 +67,7 @@ custom_theme = gr.themes.Soft(
     block_title_text_weight="600",
 )
 
-with gr.Blocks(theme=custom_theme, fill_height=True) as demo:
+with gr.Blocks(fill_height=True) as demo:
     gr.Markdown(
         """
         # 🎨 Z-Image-Turbo
@@ -168,6 +168,7 @@ with gr.Blocks(theme=custom_theme, fill_height=True) as demo:
 
 if __name__ == "__main__":
     demo.launch(
+        theme=custom_theme,
         css="""
         .header-text h1 {
             font-size: 2.5rem !important;