Update app.py

app.py

@@ -115,15 +115,22 @@ with gr.Blocks() as demo:
     #     concurrency_limit=None
     # )
 
-    generateBtn.click(
-        fn=generate_image,
-        inputs=[prompt, seed, width, height, randomize_seed, num_inference_steps],
+    gr.on(
+        triggers=[generateBtn.click, prompt.submit],
+        fn=infer,
+         inputs=[prompt, seed, width, height, randomize_seed, num_inference_steps],
         outputs=[result, seed, latency],
-        show_progress="full",
-        api_name="RealtimeFlux",
-        queue=False
     )
 
+    # generateBtn.click(
+    #     fn=generate_image,
+    #     inputs=[prompt, seed, width, height, randomize_seed, num_inference_steps],
+    #     outputs=[result, seed, latency],
+    #     show_progress="full",
+    #     api_name="RealtimeFlux",
+    #     queue=False
+    # )
+
     # def update_ui(realtime_enabled):
     #     return {
     #         prompt: gr.update(interactive=True),