Update app.py

app.py

@@ -12,7 +12,6 @@ import psutil
 import GPUtil
 import queue
 import threading
-import pandas as pd
 
 os.environ["GRADIO_TEMP_DIR"] = os.path.join(os.getcwd(), ".tmp_outputs")
 
@@ -173,7 +172,7 @@ def add_task(function, args, task_name):
 
 def get_task_status():
     with lock:
-        return pd.DataFrame([{'ID': task['id'], 'Task': task['name'], 'Status': task['status']} for task in tasks])
+        return [{'id': task['id'], 'name': task['name'], 'status': task['status']} for task in tasks]
 
 css = """
 body {
@@ -338,25 +337,23 @@ with gr.Blocks(css=css) as demo:
     generate_button.click(
         generate_vanilla,
         inputs=[prompt, num_inference_steps, guidance_scale],
-        outputs=[task_status],
-        concurrency_limit=1
+        outputs=[task_status]
     )
 
     generate_button_vs.click(
         generate_vs,
         inputs=[prompt, num_inference_steps, guidance_scale, pab_threshold, pab_gap],
-        outputs=[task_status],
-        concurrency_limit=1
+        outputs=[task_status]
     )
 
-    enhance_button.click(enhance_prompt_func, inputs=[prompt], outputs=[prompt], concurrency_limit=1)
+    enhance_button.click(enhance_prompt_func, inputs=[prompt], outputs=[prompt])
 
-    refresh_button.click(update_server_status, outputs=[cpu_status, memory_status, disk_status, gpu_status], concurrency_limit=1)
+    refresh_button.click(update_server_status, outputs=[cpu_status, memory_status, disk_status, gpu_status])
     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], concurrency_limit=1)
+    refresh_tasks_button.click(update_task_status, outputs=[task_status])
     demo.load(update_task_status, outputs=[task_status], every=5)  # 每5秒自动刷新一次
 
 if __name__ == "__main__":
-    demo.queue(max_size=10)
-    demo.launch(max_threads=4)
+    demo.queue(max_size=10, concurrency_count=1)
+    demo.launch()