fffiloni commited on
Commit
98b60eb
1 Parent(s): 0068a8e

Update gradio_app.py

Browse files
Files changed (1) hide show
  1. gradio_app.py +1 -2
gradio_app.py CHANGED
@@ -23,7 +23,6 @@ snapshot_download(
23
  'vae/**',
24
  ],
25
  local_dir=weights_dir,
26
- local_dir_use_symlinks=False,
27
  )
28
 
29
 
@@ -93,7 +92,7 @@ def single_inference(user_prompt, save_path, guidance_scale, num_sampling_steps,
93
 
94
 
95
  # Gradio interface function
96
- def run_inference(user_prompt, guidance_scale, num_sampling_steps, seed, enable_cpu_offload):
97
  save_path = "./output_videos/generated_video.mp4"
98
  result_path = single_inference(user_prompt, save_path, guidance_scale, num_sampling_steps, seed, enable_cpu_offload)
99
  return result_path
 
23
  'vae/**',
24
  ],
25
  local_dir=weights_dir,
 
26
  )
27
 
28
 
 
92
 
93
 
94
  # Gradio interface function
95
+ def run_inference(user_prompt, guidance_scale, num_sampling_steps, seed, enable_cpu_offload, progress=gr.Progress(track_tqdm=True)):
96
  save_path = "./output_videos/generated_video.mp4"
97
  result_path = single_inference(user_prompt, save_path, guidance_scale, num_sampling_steps, seed, enable_cpu_offload)
98
  return result_path