orderlymirror commited on
Commit
0d60e48
Β·
verified Β·
1 Parent(s): 118eb66

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -24,13 +24,13 @@ def generate_video(prompt: str) -> str:
24
  # Offloading moves weights automatically; just call the pipeline
25
  output = pipe(
26
  prompt=prompt,
27
- num_inference_steps=50,
28
- num_frames=161
29
  )
30
  frames = output.frames[0]
31
 
32
  # Export frames to MP4 (H.264) for browser playback
33
- return export_to_video(frames, "generated.mp4", fps=16)
34
 
35
  # 3. Build the Gradio interface
36
  with gr.Blocks(title="CogVideoX Text-to-Video") as demo:
 
24
  # Offloading moves weights automatically; just call the pipeline
25
  output = pipe(
26
  prompt=prompt,
27
+ num_inference_steps=3,
28
+ num_frames=50
29
  )
30
  frames = output.frames[0]
31
 
32
  # Export frames to MP4 (H.264) for browser playback
33
+ return export_to_video(frames, "generated.mp4", fps=15)
34
 
35
  # 3. Build the Gradio interface
36
  with gr.Blocks(title="CogVideoX Text-to-Video") as demo: