Hugging Face's logo

Spaces:
Moon11111
/
Avatar2.0
Build error

App Files Community
Spanicin commited on
Commit
fb7c8ee
·
verified ·
1 Parent(s): a40cff4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ VIDEO_DIRECTORY = None
24
 
25
  def clear_cuda_memory():
26
  torch.cuda.empty_cache()
27
- gc.collect(
28
 
29
  def run_inference(video_path, audio_path, video_out_path,
30
  inference_ckpt_path, unet_config_path="configs/unet/second_stage.yaml",
 
24
 
25
  def clear_cuda_memory():
26
  torch.cuda.empty_cache()
27
+ gc.collect()
28
 
29
  def run_inference(video_path, audio_path, video_out_path,
30
  inference_ckpt_path, unet_config_path="configs/unet/second_stage.yaml",