Update app.py

app.py

@@ -1458,7 +1458,7 @@ def handle_generation(h,w,d):
         Negative: {neg}
     """)
 
-    return pipe_generate_image(pos,neg,h,w)
+    yield pipe_generate_image(pos,neg,h,w)
     
 # entry
 
@@ -1503,7 +1503,8 @@ if __name__ == "__main__":
                 cover = gr.Image(interactive=False,container=False,elem_classes="image-container", label="Result", show_label=True, type='pil', show_share_button=False)
                 upscale_now = gr.Button("Upscale",elem_classes="btn")
 
-        run.click(
+        gr.on(
+            triggers=[run.click],
             fn=handle_generation,
             inputs=[height,width,data],
             outputs=[cover]