Update app.py

app.py

@@ -14,8 +14,8 @@ from videosys.models.cogvideo.pipeline import CogVideoPABConfig
 import psutil
 import GPUtil
 
-import threading
-task_running = threading.Event()
+# import threading
+# task_running = threading.Event()
 
 
 
@@ -263,16 +263,16 @@ with gr.Blocks(css=css) as demo:
 
 
 
-    def run_task(func, *args):
-        if task_running.is_set():
-            return None, gr.update(value="Please wait for another running end", visible=True), gr.update(visible=False)
+    # def run_task(func, *args):
+    #     if task_running.is_set():
+    #         return None, gr.update(value="Please wait for another running end", visible=True), gr.update(visible=False)
         
-        task_running.set()
-        try:
-            result = func(*args)
-            return result, gr.update(visible=False), gr.update(visible=True)
-        finally:
-            task_running.clear()
+    #     task_running.set()
+    #     try:
+    #         result = func(*args)
+    #         return result, gr.update(visible=False), gr.update(visible=True)
+    #     finally:
+    #         task_running.clear()
 
     def generate_vanilla(prompt, num_inference_steps, guidance_scale, progress=gr.Progress(track_tqdm=True)):
         engine = load_model()
@@ -306,7 +306,7 @@ with gr.Blocks(css=css) as demo:
         try:
             gpus = GPUtil.getGPUs()
             if gpus:
-                gpu = gpus[0]  # 只获取第一个GPU的信息
+                gpu = gpus[0]  
                 gpu_memory = f"{gpu.memoryUsed}/{gpu.memoryTotal}MB ({gpu.memoryUtil*100:.1f}%)"
             else:
                 gpu_memory = "No GPU found"
@@ -332,25 +332,18 @@ with gr.Blocks(css=css) as demo:
 
 
     generate_button.click(
-        run_task,
-        inputs=[gr.Textbox(value=str(generate_vanilla), visible=False), prompt, num_inference_steps, guidance_scale],
-        outputs=[
-            gr.Group([video_output, download_video_button, elapsed_time]), 
-            task_status,
-            gr.Group([generate_button, generate_button_vs])
-        ]
+        generate_vanilla,
+        inputs=[prompt, num_inference_steps, guidance_scale],
+        outputs=[video_output, download_video_button, elapsed_time],
     )
     
     generate_button_vs.click(
-        run_task,
-        inputs=[gr.Textbox(value=str(generate_vs), visible=False), prompt, num_inference_steps, guidance_scale, pab_threshold, pab_gap],
-        outputs=[
-            gr.Group([video_output_vs, download_video_button_vs, elapsed_time_vs]), 
-            task_status,
-            gr.Group([generate_button, generate_button_vs])
-        ]
+        generate_vs,
+        inputs=[prompt, num_inference_steps, guidance_scale, pab_threshold, pab_gap],
+        outputs=[video_output_vs, download_video_button_vs, elapsed_time_vs],
     )
 
+
     enhance_button.click(enhance_prompt_func, inputs=[prompt], outputs=[prompt])