Kvikontent commited on
Commit
91cab35
1 Parent(s): a50a2f9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -13,8 +13,8 @@ pipe.enable_vae_slicing()
13
 
14
  @spaces.GPU(duration=250)
15
  def generate(prompt):
16
- video_frames = pipe(prompt, num_inference_steps=25, num_frames=10).frames
17
- video_path = export_to_video(video_frames)
18
  return video_path
19
 
20
  prompt = gr.Textbox(label="Enter prompt to generate a video", info="Based on this prompt ai will generate a video")
 
13
 
14
  @spaces.GPU(duration=250)
15
  def generate(prompt):
16
+ video_frames = pipe(prompt, num_inference_steps=25, num_frames=10).frames[0]
17
+ video_path = export_to_video(video_frames, fps=10)
18
  return video_path
19
 
20
  prompt = gr.Textbox(label="Enter prompt to generate a video", info="Based on this prompt ai will generate a video")