Update app.py

app.py

@@ -95,7 +95,7 @@ with gr.Blocks(fill_width=True, css=css) as demo:
         img_width = gr.Slider(label="Image Width", minimum=64, maximum=2048, value=512, step=64)
         img_height = gr.Slider(label="Image Height", minimum=64, maximum=2048, value=1024, step=64)
         generate_button = gr.Button("Generate Image")
-        
+        output_image = gr.Image(label="output")
 
     gr.on(
         triggers=[run_base.click],