Spaces:
Runtime error
Runtime error
Update gradio_app.py
Browse files- gradio_app.py +1 -0
gradio_app.py
CHANGED
@@ -250,6 +250,7 @@ def get_image(image, prompt, steps=50, cfg_scale=7.5, eta=1.0, fs=3, seed=123, i
|
|
250 |
if len(prompt_str) == 0:
|
251 |
prompt_str = 'empty_prompt'
|
252 |
|
|
|
253 |
save_videos(batch_samples, result_dir, filenames=[prompt_str], fps=save_fps)
|
254 |
print(f"Saved in {prompt_str}. Time used: {(time.time() - start):.2f} seconds")
|
255 |
model = model.cpu()
|
|
|
250 |
if len(prompt_str) == 0:
|
251 |
prompt_str = 'empty_prompt'
|
252 |
|
253 |
+
global result_dir
|
254 |
save_videos(batch_samples, result_dir, filenames=[prompt_str], fps=save_fps)
|
255 |
print(f"Saved in {prompt_str}. Time used: {(time.time() - start):.2f} seconds")
|
256 |
model = model.cpu()
|