Update app.py

app.py

@@ -115,6 +115,7 @@ pipe = SimpleDiffusionPipeline(
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 640
+MIN_IMAGE_SIZE = 320
 
 @spaces.GPU #[uncomment to use ZeroGPU]
 def infer(
@@ -205,7 +206,7 @@ with gr.Blocks(css=css) as demo:
             with gr.Row():
                 width = gr.Slider(
                     label="Width",
-                    minimum=384,
+                    minimum=MIN_IMAGE_SIZE,
                     maximum=MAX_IMAGE_SIZE,
                     step=64,
                     value=512,  # Replace with defaults that work for your model
@@ -213,7 +214,7 @@ with gr.Blocks(css=css) as demo:
 
                 height = gr.Slider(
                     label="Height",
-                    minimum=384,
+                    minimum=MIN_IMAGE_SIZE,
                     maximum=MAX_IMAGE_SIZE,
                     step=64,
                     value=640,  # Replace with defaults that work for your model