fbnnb commited on
Commit
9dec6c1
·
verified ·
1 Parent(s): fe89f07

Update gradio_app.py

Browse files
Files changed (1) hide show
  1. gradio_app.py +3 -3
gradio_app.py CHANGED
@@ -258,9 +258,9 @@ def get_image(image, prompt, steps=50, cfg_scale=7.5, eta=1.0, fs=3, seed=123, i
258
  save_videos(batch_samples, result_dir, filenames=[prompt_str], fps=save_fps)
259
  print(f"Saved in {prompt_str}. Time used: {(time.time() - start):.2f} seconds")
260
  model = model.cpu()
261
- result_dir = os.path.join(result_dir, f"{prompt_str}.mp4")
262
- print("result saved to:", result_dir)
263
- return result_dir
264
 
265
 
266
  # @spaces.GPU
 
258
  save_videos(batch_samples, result_dir, filenames=[prompt_str], fps=save_fps)
259
  print(f"Saved in {prompt_str}. Time used: {(time.time() - start):.2f} seconds")
260
  model = model.cpu()
261
+ saved_result_dir = os.path.join(result_dir, f"{prompt_str}.mp4")
262
+ print("result saved to:", saved_result_dir)
263
+ return saved_result_dir
264
 
265
 
266
  # @spaces.GPU