Update app.py

app.py

@@ -225,23 +225,28 @@ HTML_TEMPLATE = """
 </div>
 """
 
-# Gradio Interface
-with gr.Blocks(css=HTML_TEMPLATE) as demo:
-    gr.HTML(HTML_TEMPLATE)
+with gr.Blocks(css=css) as demo:
+    gr.HTML("<div class='header'><h1>🌟 Dreamscape Creator</h1></div>")
+    
     with gr.Row():
         text = gr.Textbox(label="Prompt", placeholder="Describe your dreamscape...")
         style = gr.Dropdown(label="Style", choices=list(style_token_dict.keys()), value="Illustration Style")
+    
     with gr.Row():
         inference_step = gr.Slider(1, 50, 20, step=1, label="Inference steps")
         guidance_scale = gr.Slider(1, 10, 7.5, step=0.1, label="Guidance scale")
         seed = gr.Slider(0, 10000, 42, step=1, label="Seed")
+    
     with gr.Row():
         guidance_method = gr.Dropdown(label="Guidance method", choices=['Grayscale', 'Bright', 'Contrast', 'Symmetry', 'Saturation'], value="Grayscale")
         loss_scale = gr.Slider(100, 10000, 200, step=100, label="Loss scale")
+    
     with gr.Row():
         image_size = gr.Radio(["256x256", "512x512"], label="Image Size", value="256x256")
+    
     with gr.Row():
         generate_button = gr.Button("Create Dreamscape", variant="primary")
+    
     with gr.Row():
         output_image = gr.Image(label="Your Dreamscape")
         output_image_guided = gr.Image(label="Guided Dreamscape")
@@ -252,24 +257,18 @@ with gr.Blocks(css=HTML_TEMPLATE) as demo:
         outputs=[output_image, output_image_guided]
     )
 
-    gr.Markdown("""
-        Note: Example generation may take some time due to CPU limitations. 
-        Thank you for your patience!
-    """)
-
-   gr.Examples(
-    examples=[
-        ["Floating island with waterfalls", 'Illustration Style', 50, 7.5, 42, 'Grayscale', 200, "256x256"],
-        ["Futuristic city with neon lights", 'Line Art', 30, 8.0, 123, 'Bright', 300, "256x256"],
-        ["Japanese garden with cherry blossoms", 'Hitokomoru Style', 40, 7.0, 789, 'Contrast', 250, "256x256"],
-    ],
-    inputs=[text, style, inference_step, guidance_scale, seed, guidance_method, loss_scale, image_size],
-    outputs=[output_image, output_image_guided],
-    fn=inference,
-    cache_examples=True,
-    examples_per_page=5,
-    cache_timeout=360  # Increase this value (in seconds) as needed
-
+    gr.Examples(
+        examples=[
+            ["Floating island with waterfalls", 'Illustration Style', 50, 7.5, 42, 'Grayscale', 200, "256x256"],
+            ["Futuristic city with neon lights", 'Line Art', 30, 8.0, 123, 'Bright', 300, "256x256"],
+            ["Japanese garden with cherry blossoms", 'Hitokomoru Style', 40, 7.0, 789, 'Contrast', 250, "256x256"],
+        ],
+        inputs=[text, style, inference_step, guidance_scale, seed, guidance_method, loss_scale, image_size],
+        outputs=[output_image, output_image_guided],
+        fn=inference,
+        cache_examples=True,
+        examples_per_page=5,
+        cache_timeout=360
     )
 
 if __name__ == "__main__":