RamAnanth1 commited on
Commit
664012c
β€’
1 Parent(s): 31cdd21

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -108,10 +108,10 @@ def save_results(videos,
108
 
109
  return os.path.join(save_subdir, f"{save_name}_{i:03d}.mp4")
110
 
111
- def get_video(prompt, seed):
112
  seed_everything(seed)
113
  samples = sample_text2video(model, prompt, n_samples = 1, batch_size = 1,
114
- sampler=ddim_sampler,
115
  )
116
  return save_results(samples)
117
 
 
108
 
109
  return os.path.join(save_subdir, f"{save_name}_{i:03d}.mp4")
110
 
111
+ def get_video(prompt, seed, ddim_steps):
112
  seed_everything(seed)
113
  samples = sample_text2video(model, prompt, n_samples = 1, batch_size = 1,
114
+ sampler=ddim_sampler, ddim_steps=ddim_steps
115
  )
116
  return save_results(samples)
117