Update app.py

app.py

@@ -117,7 +117,7 @@ with gr.Blocks() as demo:
 
     gr.on(
         triggers=[generateBtn.click, prompt.submit],
-        fn=infer,
+        fn=generate_image,
          inputs=[prompt, seed, width, height, randomize_seed, num_inference_steps],
         outputs=[result, seed, latency],
     )