fbnnb commited on
Commit
002e0fb
·
verified ·
1 Parent(s): 02796fb

Update gradio_app.py

Browse files
Files changed (1) hide show
  1. 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()