Update app.py

app.py

@@ -40,6 +40,7 @@ def generate_image(prompt, option, progress=gr.Progress()):
         step_loaded = step
     def inference_callback(p, i, t, kwargs):
         progress(i+1, total=step)
+        return kwargs
     return pipe(prompt, num_inference_steps=step, guidance_scale=0, callback_on_step_end=inference_callback).images[0]
 
 with gr.Blocks(css="style.css") as demo: