Update app.py

app.py

@@ -338,23 +338,25 @@ with gr.Blocks(css=css) as demo:
     generate_button.click(
         generate_vanilla,
         inputs=[prompt, num_inference_steps, guidance_scale],
-        outputs=[task_status]
+        outputs=[task_status],
+        concurrency_limit=1
     )
 
     generate_button_vs.click(
         generate_vs,
         inputs=[prompt, num_inference_steps, guidance_scale, pab_threshold, pab_gap],
-        outputs=[task_status]
+        outputs=[task_status],
+        concurrency_limit=1
     )
 
-    enhance_button.click(enhance_prompt_func, inputs=[prompt], outputs=[prompt])
+    enhance_button.click(enhance_prompt_func, inputs=[prompt], outputs=[prompt], concurrency_limit=1)
 
-    refresh_button.click(update_server_status, outputs=[cpu_status, memory_status, disk_status, gpu_status])
+    refresh_button.click(update_server_status, outputs=[cpu_status, memory_status, disk_status, gpu_status], concurrency_limit=1)
     demo.load(update_server_status, outputs=[cpu_status, memory_status, disk_status, gpu_status], every=1)
 
-    refresh_tasks_button.click(update_task_status, outputs=[task_status])
+    refresh_tasks_button.click(update_task_status, outputs=[task_status], concurrency_limit=1)
     demo.load(update_task_status, outputs=[task_status], every=5)  # 每5秒自动刷新一次
 
 if __name__ == "__main__":
-    demo.queue(max_size=10, concurrency_count=1)
-    demo.launch()
+    demo.queue(max_size=10)
+    demo.launch(max_threads=4)