Spaces:
Runtime error
Runtime error
RamAnanth1
commited on
Commit
β’
664012c
1
Parent(s):
31cdd21
Update app.py
Browse files
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 |
|