Update app.py

app.py

@@ -17,21 +17,21 @@ KEEP_RUNING_TASK = 3600 * 2
 # the total running task number in 1800 seconds
 LIMIT_RUNING_TASK = 10
 
-def t2v_generation(prompt, resolution, watermark_wanx, seed = -1):
+def t2v_generation(prompt, resolution, watermark_wan, seed = -1):
     seed = seed if seed >= 0 else random.randint(0, 2147483647)
     if not allow_task_num():
         gr.Info(f"Warning: The number of running tasks is too large, please wait for a while.")
         return None, gr.Button(visible=True)
     try:
         rsp = VideoSynthesis.call(model="wanx2.1-t2v-plus", prompt=prompt, seed=seed,
-                                  watermark_wanx=watermark_wanx, size=resolution)
+                                  watermark_wanx=watermark_wan, size=resolution)
         video_url = rsp.output.video_url
         return video_url, gr.Button(visible=True)
     except Exception as e:
         gr.Warning(f"Warning: {e}")
         return None, gr.Button(visible=True)
 
-def t2v_generation_async(prompt, size, watermark_wanx, seed = -1):
+def t2v_generation_async(prompt, size, watermark_wan, seed = -1):
     print(seed)
     seed = seed if seed >= 0 else random.randint(0, 2147483647)
     print(seed)
@@ -43,7 +43,7 @@ def t2v_generation_async(prompt, size, watermark_wanx, seed = -1):
                                         prompt=prompt,
                                         size=size,
                                         seed=seed,
-                                        watermark_wanx=watermark_wanx)
+                                        watermark_wanx=watermark_wan)
         task_id = rsp.output.task_id
         status = False
         return task_id, status, gr.Button(visible=False)
@@ -51,27 +51,27 @@ def t2v_generation_async(prompt, size, watermark_wanx, seed = -1):
         gr.Warning(f"Warning: {e}")
         return None, True, gr.Button()
 
-def i2v_generation(prompt, image, watermark_wanx, seed = -1):
+def i2v_generation(prompt, image, watermark_wan, seed = -1):
     seed = seed if seed >= 0 else random.randint(0, 2147483647)
     video_url = None
     try:
         rsp = VideoSynthesis.call(model="wanx2.1-i2v-plus", prompt=prompt, img_url= image,
                                   seed = seed,
-                                  watermark_wanx=watermark_wanx
+                                  watermark_wanx=watermark_wan
                                   )
         video_url = rsp.output.video_url
     except Exception as e:
         gr.Warning(f"Warning: {e}")
     return video_url
 
-def i2v_generation_async(prompt, image, watermark_wanx, seed = -1):
+def i2v_generation_async(prompt, image, watermark_wan, seed = -1):
     seed = seed if seed >= 0 else random.randint(0, 2147483647)
     if not allow_task_num():
         gr.Info(f"Warning: The number of running tasks is too large, please wait for a while.")
         return "", None, gr.Button(visible=True)
     try:
         rsp = VideoSynthesis.async_call(model="wanx2.1-i2v-plus", prompt=prompt, seed=seed,
-                                        img_url= image, watermark_wanx=watermark_wanx)
+                                        img_url= image, watermark_wanx=watermark_wan)
         print(rsp)
         task_id = rsp.output.task_id
         status = False
@@ -206,7 +206,7 @@ with gr.Blocks() as demo:
                                           interactive=False,
                                           height=500)
             with gr.Row():
-                    watermark_wanx = gr.Checkbox(label="Watermark", value=True, container=False)
+                    watermark_wan = gr.Checkbox(label="Watermark", value=True, container=False)
                     seed = gr.Number(label="Seed", value=-1, container=True)
                     cost_time = gr.Number(label="Cost Time(secs)", value=cost_time, interactive=False,
                                            every=2, inputs=[task_id], container=True)
@@ -248,13 +248,13 @@ with gr.Blocks() as demo:
 
     run_t2v_button.click(
         fn=t2v_generation_async,
-        inputs=[txt2vid_prompt, resolution, watermark_wanx, seed],
+        inputs=[txt2vid_prompt, resolution, watermark_wan, seed],
         outputs=[task_id, status, run_t2v_button],
     )
 
     run_i2v_button.click(
         fn=i2v_generation_async,
-        inputs=[img2vid_prompt, img2vid_image, watermark_wanx, seed],
+        inputs=[img2vid_prompt, img2vid_image, watermark_wan, seed],
         outputs=[task_id, status, run_i2v_button],
     )