Update Modules/Generate_Video.py

Modules/Generate_Video.py

@@ -54,7 +54,7 @@ def _write_video_tmp(data_iter_or_bytes: object, suffix: str = ".mp4") -> str:
 )
 def Generate_Video(
     prompt: Annotated[str, "Text description of the video to generate (e.g., 'a red fox running through a snowy forest at sunrise')."],
-    model_id: Annotated[str, "Hugging Face model id in the form 'creator/model-name'. Defaults to akhaliq/sora-2."] = "akhaliq/sora-2",
+    model_id: Annotated[str, "Hugging Face model id in the form 'creator/model-name'. Defaults to Wan-AI/Wan2.2-T2V-A14B."] = "Wan-AI/Wan2.2-T2V-A14B",
     negative_prompt: Annotated[str, "What should NOT appear in the video."] = "",
     steps: Annotated[int, "Number of denoising steps (1–100). Higher can improve quality but is slower."] = 25,
     cfg_scale: Annotated[float, "Guidance scale (1–20). Higher = follow the prompt more closely, lower = more creative."] = 3.5,
@@ -62,7 +62,7 @@ def Generate_Video(
     width: Annotated[int, "Output width in pixels (multiples of 8 recommended)."] = 768,
     height: Annotated[int, "Output height in pixels (multiples of 8 recommended)."] = 768,
     fps: Annotated[int, "Frames per second of the output video (e.g., 24)."] = 24,
-    duration: Annotated[float, "Target duration in seconds. For Sora-2, must be 4, 8, or 12."] = 4.0,
+    duration: Annotated[float, "Target duration in seconds, default 4."] = 4.0,
 ) -> str:
     _log_call_start(
         "Generate_Video",
@@ -127,7 +127,7 @@ def Generate_Video(
         except KeyError as exc:
             # Handle specific provider errors that manifest as KeyError (e.g. fal-ai missing 'video' key on error)
             if "video" in str(exc):
-                last_error = ValueError(f"Provider {provider} returned an invalid response. This often happens with invalid parameters (e.g. duration must be 4, 8, or 12 for Sora-2).")
+                last_error = ValueError(f"Provider {provider} returned an invalid response. This often happens with invalid parameters.")
             else:
                 last_error = exc
             continue
@@ -146,7 +146,7 @@ def build_interface() -> gr.Interface:
             gr.Textbox(label="Prompt", placeholder="Enter a prompt for the video", lines=2, info="Text description of the video to generate"),
             gr.Textbox(
                 label="Model",
-                value="akhaliq/sora-2",
+                value="Wan-AI/Wan2.2-T2V-A14B",
                 placeholder="creator/model-name",
                 max_lines=1,
                 info="<a href=\"https://huggingface.co/models?pipeline_tag=text-to-video&inference_provider=nebius,cerebras,novita,fireworks-ai,together,fal-ai,groq,featherless-ai,nscale,hyperbolic,sambanova,cohere,replicate,scaleway,publicai,hf-inference&sort=trending\" target=\"_blank\" rel=\"noopener noreferrer\">Browse models</a>",
@@ -164,7 +164,7 @@ def build_interface() -> gr.Interface:
         title="Generate Video",
         description=(
             "<div style=\"text-align:center\">Generate short videos via Hugging Face serverless inference. "
-            "Default model is Sora-2.</div>"
+            "Default model is Wan2.2-T2V-A14B.</div>"
         ),
         api_description=TOOL_SUMMARY,
         flagging_mode="never",