Update app.py

app.py

@@ -25,6 +25,8 @@ with gr.Blocks() as demo:
             num_samples = gr.Number(label="Number of Samples", value=4)
             guidance_scale = gr.Number(label="Guidance Scale", value=7.5)
     with gr.Row():
+        height = gr.Number(label="Height", value=512)
+        width = gr.Number(label="Width", value=512)
         num_inference_steps = gr.Slider(label="Steps", value=24)
     gallery = gr.Gallery()