Update app.py

app.py

@@ -15,9 +15,6 @@ translator = Translator()
 model = "black-forest-labs/FLUX.1-dev"
 
 CSS = """
-.gradio-container {
-  max-width: 690px !important;
-}
 footer {
     visibility: hidden;
 }
@@ -85,7 +82,7 @@ examples = [
 
 # Gradio Interface
 
-with gr.Blocks(css=CSS, theme="Citrus") as demo:
+with gr.Blocks(css=CSS, theme="ocean") as demo:
     gr.HTML("<h1><center>flux.1-lumiere</center></h1>")
     gr.HTML("<p><center><a href='https://huggingface.co/aixonlab/flux.1-lumiere-alpha</a></center></p>")
     with gr.Group():
@@ -146,7 +143,8 @@ with gr.Blocks(css=CSS, theme="Citrus") as demo:
         inputs=prompt,
         outputs=[img,seed],
         fn=generate_image,
-        cache_examples="lazy",
+        cache_examples=True,
+        cache_mode='lazy'
     )
 
     gr.on(