Update app.py

app.py

@@ -31,8 +31,8 @@ class SimpleDiffusionPipeline(DiffusionPipeline):
         self,
         prompt,
         negative_prompt=None,
-        height=384,
-        width=384,
+        height=640,
+        width=640,
         num_inference_steps=40,
         guidance_scale=5.5,
         generator=None,
@@ -114,7 +114,7 @@ pipe = SimpleDiffusionPipeline(
 ).to(device)
 
 MAX_SEED = np.iinfo(np.int32).max
-MAX_IMAGE_SIZE = 384
+MAX_IMAGE_SIZE = 640
 
 @spaces.GPU #[uncomment to use ZeroGPU]
 def infer(
@@ -205,18 +205,18 @@ with gr.Blocks(css=css) as demo:
             with gr.Row():
                 width = gr.Slider(
                     label="Width",
-                    minimum=192,
+                    minimum=384,
                     maximum=MAX_IMAGE_SIZE,
                     step=64,
-                    value=256,  # Replace with defaults that work for your model
+                    value=512,  # Replace with defaults that work for your model
                 )
 
                 height = gr.Slider(
                     label="Height",
-                    minimum=192,
+                    minimum=384,
                     maximum=MAX_IMAGE_SIZE,
                     step=64,
-                    value=384,  # Replace with defaults that work for your model
+                    value=640,  # Replace with defaults that work for your model
                 )
 
             with gr.Row():