Update app.py

app.py

@@ -518,7 +518,7 @@ def main():
                 queue=False)
 
             gen_btn.click(
-                fn=partial(click_on_display),
+                fn=partial(generate, unet, vae, tokenizer, text_encoder,),
                 inputs=[
                     language_instruction, grounding_instruction, sketch_pad,
                     loss_threshold, guidance_scale, batch_size, rand_seed,