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

Update gradio_app.py

Browse files
Files changed (1) hide show
  1. gradio_app.py +1 -0
gradio_app.py CHANGED
@@ -251,6 +251,7 @@ def get_image(image, prompt, steps=50, cfg_scale=7.5, eta=1.0, fs=3, seed=123, i
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()
 
251
  prompt_str = 'empty_prompt'
252
 
253
  global result_dir
254
+ global save_fps
255
  save_videos(batch_samples, result_dir, filenames=[prompt_str], fps=save_fps)
256
  print(f"Saved in {prompt_str}. Time used: {(time.time() - start):.2f} seconds")
257
  model = model.cpu()