update app

app.py

@@ -614,6 +614,10 @@ def create_chatbot_demo():
         test_inputs=[prefix_input, output_vis, suffix_input, test_case_input]
 
         generate_btn.click(
+            fn=lambda: ("Waiting for ZeroGPU...", ''),
+            outputs=[output_vis, result_output],
+            show_progress="hidden"
+        ).then(
             fn=infilling_dream,
             inputs=generation_inputs,
             outputs=[output_vis, result_output],