Update app.py

app.py

@@ -11,7 +11,7 @@ import torch
 from diffusers import StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
 
 # Use environment variables for flexibility
-MODEL_ID = os.getenv("MODEL_ID", "Corcelio/mobius")
+MODEL_ID = os.getenv("MODEL_ID", "sd-community/sdxl-flash")
 MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "4096"))
 USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE", "0") == "1"
 ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
@@ -382,7 +382,7 @@ with gr.Blocks(css=css) as demo:
             generate_btn = gr.Button("Generate")
 
     with gr.Row():
-        image_output = gr.Gallery(label="Generated Images", grid=[2], height="auto")
+        image_output = gr.Gallery(label="Generated Images", columns=2, height="auto")
         seed_output = gr.Number(label="Seed Used")
 
     gr.Examples(examples=examples, inputs=[prompt])
@@ -395,4 +395,4 @@ with gr.Blocks(css=css) as demo:
 
     generate_btn.click(fn=generate, inputs=inputs, outputs=[image_output, seed_output])
 
-demo.launch()
+demo.launch()