Update app.py

app.py

@@ -84,7 +84,7 @@ with gr.Blocks() as demo:
                     button = gr.Button("Generate", variant="primary")
                 with gr.Column():
                     output = gr.Gallery(label='Generated Images', height=700)
-                    button.click(demo_util.sample_fn, inputs=[
+                    button.click(demo_infer, inputs=[
                         guidance_scale, randomize_temperature, num_sample_steps,
                         i1k_class, seed],
                         outputs=[output])