Update app.py

app.py

@@ -217,7 +217,7 @@ if __name__ == '__main__':
                 status_textbox = gr.Textbox(label="Status", interactive=False)
         
         submit_button.click(
-            fn=ui_wrapper,
+            fn=generate,
             inputs=input_textbox,
             outputs=[output_textbox, status_textbox]
         )