Update app.py

app.py

@@ -14,9 +14,6 @@ from videosys.models.cogvideo.pipeline import CogVideoPABConfig
 import psutil
 import GPUtil
 
-# import threading
-# task_running = threading.Event()
-
 
 
 logging.basicConfig(level=logging.INFO)
@@ -245,23 +242,6 @@ with gr.Blocks(css=css) as demo:
                         disk_status = gr.Textbox(label="Disk", scale=1)
                     gpu_status = gr.Textbox(label="GPU Memory", scale=1)
                     refresh_button = gr.Button("Refresh", scale=1, size="sm")
-                with gr.Column(elem_classes="server-status"):
-                    gr.Markdown("#### Server Status")
-                    
-                    with gr.Row():
-                        with gr.Column(scale=1):
-                            cpu_status = gr.Textbox(label="CPU")
-                        with gr.Column(scale=1):
-                            memory_status = gr.Textbox(label="Memory")
-                    
-                    with gr.Row():
-                        with gr.Column(scale=1):
-                            disk_status = gr.Textbox(label="Disk")
-                        with gr.Column(scale=1):
-                            gpu_status = gr.Textbox(label="GPU Memory")
-                    
-                    with gr.Row():
-                        refresh_button = gr.Button("Refresh", size="sm")
 
         with gr.Column():
             with gr.Row():
@@ -278,18 +258,7 @@ with gr.Blocks(css=css) as demo:
             task_status = gr.Textbox(label="任务状态", 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()
+         
 
     def generate_vanilla(prompt, num_inference_steps, guidance_scale, progress=gr.Progress(track_tqdm=True)):
         engine = load_model()
@@ -298,7 +267,7 @@ with gr.Blocks(css=css) as demo:
         elapsed_time = time() - t
         video_update = gr.update(visible=True, value=video_path)
         elapsed_time = gr.update(visible=True, value=f"{elapsed_time:.2f}s")
-    
+
         return video_path, video_update, elapsed_time
 
     def generate_vs(prompt, num_inference_steps, guidance_scale, threshold, gap, progress=gr.Progress(track_tqdm=True)):
@@ -310,7 +279,7 @@ with gr.Blocks(css=css) as demo:
         elapsed_time = time() - t
         video_update = gr.update(visible=True, value=video_path)
         elapsed_time = gr.update(visible=True, value=f"{elapsed_time:.2f}s")
-    
+
         return video_path, video_update, elapsed_time
 
     def enhance_prompt_func(prompt):
@@ -323,7 +292,7 @@ with gr.Blocks(css=css) as demo:
         try:
             gpus = GPUtil.getGPUs()
             if gpus:
-                gpu = gpus[0]  
+                gpu = gpus[0]  # 只获取第一个GPU的信息
                 gpu_memory = f"{gpu.memoryUsed}/{gpu.memoryTotal}MB ({gpu.memoryUtil*100:.1f}%)"
             else:
                 gpu_memory = "No GPU found"
@@ -347,20 +316,19 @@ with gr.Blocks(css=css) as demo:
             status['gpu_memory']
         )
 
-
+    
     generate_button.click(
         generate_vanilla,
         inputs=[prompt, num_inference_steps, guidance_scale],
         outputs=[video_output, download_video_button, elapsed_time],
     )
-    
+
     generate_button_vs.click(
         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])
 
     
@@ -369,4 +337,4 @@ with gr.Blocks(css=css) as demo:
 
 if __name__ == "__main__":
     demo.queue(max_size=10, default_concurrency_limit=1)
-    demo.launch()
+    demo.launch()