Nymbo commited on
Commit
56e0bbb
·
verified ·
1 Parent(s): 50805f0

Changing default video gen model from Wan2.2-T2V-A14B to Sora-2

Browse files
Files changed (1) hide show
  1. Modules/Generate_Video.py +3 -3
Modules/Generate_Video.py CHANGED
@@ -51,7 +51,7 @@ def _write_video_tmp(data_iter_or_bytes: object, suffix: str = ".mp4") -> str:
51
  )
52
  def Generate_Video(
53
  prompt: Annotated[str, "Text description of the video to generate (e.g., 'a red fox running through a snowy forest at sunrise')."],
54
- 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",
55
  negative_prompt: Annotated[str, "What should NOT appear in the video."] = "",
56
  steps: Annotated[int, "Number of denoising steps (1–100). Higher can improve quality but is slower."] = 25,
57
  cfg_scale: Annotated[float, "Guidance scale (1–20). Higher = follow the prompt more closely, lower = more creative."] = 3.5,
@@ -146,7 +146,7 @@ def build_interface() -> gr.Interface:
146
  gr.Textbox(label="Prompt", placeholder="Enter a prompt for the video", lines=2),
147
  gr.Textbox(
148
  label="Model",
149
- value="Wan-AI/Wan2.2-T2V-A14B",
150
  placeholder="creator/model-name",
151
  max_lines=1,
152
  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:
164
  title="Generate Video",
165
  description=(
166
  "<div style=\"text-align:center\">Generate short videos via Hugging Face serverless inference. "
167
- "Default model is Wan2.2-T2V-A14B.</div>"
168
  ),
169
  api_description=TOOL_SUMMARY,
170
  flagging_mode="never",
 
51
  )
52
  def Generate_Video(
53
  prompt: Annotated[str, "Text description of the video to generate (e.g., 'a red fox running through a snowy forest at sunrise')."],
54
+ model_id: Annotated[str, "Hugging Face model id in the form 'creator/model-name'. Defaults to akhaliq/sora-2."] = "akhaliq/sora-2",
55
  negative_prompt: Annotated[str, "What should NOT appear in the video."] = "",
56
  steps: Annotated[int, "Number of denoising steps (1–100). Higher can improve quality but is slower."] = 25,
57
  cfg_scale: Annotated[float, "Guidance scale (1–20). Higher = follow the prompt more closely, lower = more creative."] = 3.5,
 
146
  gr.Textbox(label="Prompt", placeholder="Enter a prompt for the video", lines=2),
147
  gr.Textbox(
148
  label="Model",
149
+ value="akhaliq/sora-2",
150
  placeholder="creator/model-name",
151
  max_lines=1,
152
  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
  title="Generate Video",
165
  description=(
166
  "<div style=\"text-align:center\">Generate short videos via Hugging Face serverless inference. "
167
+ "Default model is Sora-2.</div>"
168
  ),
169
  api_description=TOOL_SUMMARY,
170
  flagging_mode="never",