Update simple_app.py

simple_app.py

@@ -184,19 +184,6 @@ def infer(prompt, video_size, frame_num, progress=gr.Progress(track_tqdm=True)):
         raise Exception("Error executing command")
 
 
-# 创建一个信息面板，告知用户处理时间
-def create_info_block():
-    return gr.Markdown("""
-    # Wan 2.1 T2V Model Information
-    
-    **Processing Time Expectations:**
-    - 10-30 frames: ~2-5 minutes
-    - 40-60 frames: ~5-10 minutes
-    - 81 frames: ~10-15 minutes
-    
-    Processing time depends on your GPU model and current system load.
-    """)
-
 with gr.Blocks() as demo:
     with gr.Column():
         gr.Markdown("# Wan 2.1 1.3B")
@@ -215,53 +202,28 @@ with gr.Blocks() as demo:
         </div>
         """
         )
-        
-        # 添加信息面板
-        info_block = create_info_block()
-        
-        with gr.Row():
-            with gr.Column(scale=3):
-                prompt = gr.Textbox(
-                    label="Prompt", 
-                    placeholder="Enter a detailed description of the video you want to generate",
-                    lines=3
-                )
-            with gr.Column(scale=2):
-                video_size_dropdown = gr.Dropdown(
-                    choices=SIZE_OPTIONS,
-                    value=SIZE_OPTIONS[0],  # 默认选择第一个分辨率
-                    label="Video Size (Resolution)",
-                )
-                frame_num_slider = gr.Slider(
-                    minimum=FRAME_NUM_OPTIONS[0],  # 最小帧数
-                    maximum=FRAME_NUM_OPTIONS[-1],  # 最大帧数
-                    value=30,  # 设置一个合理的默认帧数，避免处理时间过长
-                    step=10,  # 增加步长，更容易选择
-                    label="Frame Number (Video Length)",
-                )
+        prompt = gr.Textbox(label="Prompt")
+
+        video_size_dropdown = gr.Dropdown(
+            choices=SIZE_OPTIONS,
+            value=SIZE_OPTIONS[0],  # 默认选择第一个分辨率
+            label="Video Size (Resolution)",
+        )
+        frame_num_slider = gr.Slider(
+            minimum=FRAME_NUM_OPTIONS[0],  # 最小帧数
+            maximum=FRAME_NUM_OPTIONS[-1],  # 最大帧数
+            value=FRAME_NUM_OPTIONS[2],  # 默认帧数 (例如，选择列表的第三个)
+            step=1,  # 步长为 1
+            label="Frame Number (Video Length)",
+        )
 
-        submit_btn = gr.Button("Generate Video", variant="primary")
-        with gr.Row():
-            with gr.Column():
-                video_res = gr.Video(label="Generated Video")
-            with gr.Column():
-                gallery = gr.Gallery(label="Preview Frames", show_label=True, elem_id="gallery").style(grid=3, height="auto")
+        submit_btn = gr.Button("Submit")
+        video_res = gr.Video(label="Generated Video")
 
     submit_btn.click(
         fn=infer,
-        inputs=[prompt, video_size_dropdown, frame_num_slider],
+        inputs=[prompt, video_size_dropdown, frame_num_slider],  # inputs 添加 video_size_dropdown, frame_num_slider
         outputs=[video_res],
-        api_name="generate"  # 添加API名称，方便调用
     )
 
-# 使用Gradio 5.20.1中的新功能来配置队列和防止超时
-demo.queue(
-    concurrency_count=1,         # 视频生成计算密集，一次只处理一个请求
-    max_size=20,                 # 队列最大容量
-    status_update_rate=10,       # 状态更新频率（秒）
-    default_concurrency_limit=1  # 默认并发限制
-).launch(
-    share=False,                 # 不公开分享
-    show_error=True,             # 显示详细错误信息
-    show_api=False              # 不显示API文档
-)
+demo.queue().launch(share=False, show_error=True, show_api=False)