Update app.py

app.py

@@ -67,7 +67,7 @@ def generate_t2v(prompt, size="1280*704", duration_seconds=5, steps=25, progress
     return dest_path, download_link, "Text-to-Video generated successfully!"
 
 @spaces.GPU(duration=get_duration)
-def generate_i2v(image, prompt, size="1280*704", duration_seconds=5, steps=25, progress=None):
+def generate_i2v(image, prompt, size="1280*704", duration_seconds=5, steps=25, progress=gr.Progress(track_tqdm=True)):
     if image is None or not prompt.strip():
         return None, None, "Please upload an image and enter a prompt."
 
@@ -105,7 +105,7 @@ def generate_i2v(image, prompt, size="1280*704", duration_seconds=5, steps=25, p
 
 # ---------------- Step 4: Gradio UI ----------------
 with gr.Blocks() as demo:
-    progress = gr.Progress(track_tqdm=True)
+   
 
     gr.Markdown("## 🎥 Wan2.2-TI2V-5B Video Generator")
     gr.Markdown("Choose **Text-to-Video** or **Image-to-Video** mode below.")
@@ -124,7 +124,7 @@ with gr.Blocks() as demo:
         t2v_status = gr.Textbox(label="Status")
         t2v_btn.click(
             generate_t2v,
-            [t2v_prompt, t2v_size, t2v_duration, t2v_steps, progress],
+            [t2v_prompt, t2v_size, t2v_duration, t2v_steps],
             [t2v_video, t2v_download, t2v_status]
         )
 
@@ -150,7 +150,7 @@ with gr.Blocks() as demo:
         i2v_status = gr.Textbox(label="Status")
         i2v_btn.click(
             generate_i2v,
-            [i2v_image, i2v_prompt, i2v_size, i2v_duration, i2v_steps, progress],
+            [i2v_image, i2v_prompt, i2v_size, i2v_duration, i2v_steps],
             [i2v_video, i2v_download, i2v_status]
         )