Update app.py

app.py

@@ -132,8 +132,12 @@ with gr.Blocks(css=CSS, theme=gr.themes.Soft()) as demo:
     inputs = [prompt, aspect_ratio, num_inference_steps, guidance_scale, seed, randomize_seed]
     outputs = [result, seed]
 
-    run_btn.click(generate, inputs=inputs, outputs=outputs)
-    prompt.submit(generate, inputs=inputs, outputs=outputs)
+    gr.on(
+        [run_btn.click, prompt.submit],
+        generate,
+        inputs=inputs,
+        outputs=outputs,
+    )
 
 if __name__ == "__main__":
     demo.queue().launch()