Update app.py

app.py

@@ -155,7 +155,7 @@ with gr.Blocks(css=css) as demo:
         ["city landscape with fancy car, racing on the road, gopro, intricate details, 4k, cyberpunk", 7, 20, 704, 704],
         ["portrait of liu yifei girl, soldier working in a cyberpunk city, cleavage, intricate, 8k, highly detailed, digital painting, intense, sharp focus", 7, 20, 704, 704],
         ["portrait of a muscular beard male in dgs illustration style, half-body, holding robot arms, strong chest", 7, 20, 704, 704],
-    ], [prompt, guidance, steps], image_out, inference_example, cache_examples=torch.cuda.is_available())
+    ], [prompt, guidance, steps, width_input, height_input], image_out, inference_example, cache_examples=torch.cuda.is_available())
     gr.Markdown('''
       Models and Space by [@DGSpitzer](https://www.youtube.com/channel/UCzzsYBF4qwtMwJaPJZ5SuPg)❤️ [@大谷的游戏创作小屋](https://space.bilibili.com/176003)
       [![Twitter Follow](https://img.shields.io/twitter/follow/DGSpitzer?label=%40DGSpitzer&style=social)](https://twitter.com/DGSpitzer)