H-Liu1997 commited on
Commit
dac4f1f
·
verified ·
1 Parent(s): 534c45b

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +3 -2
inference.py CHANGED
@@ -1,3 +1,6 @@
 
 
 
1
  import os
2
  import gc
3
  import soundfile as sf
@@ -19,7 +22,6 @@ import smplx
19
  import igraph
20
  import fire
21
 
22
- from utils.download_utils import download_files_from_repo
23
  from utils.video_io import save_videos_from_pil
24
  from utils.genextend_inference_utils import adjust_statistics_to_match_reference
25
  from create_graph import path_visualization, graph_pruning, get_motion_reps_tensor, path_visualization_v2
@@ -526,7 +528,6 @@ OUTPUT_DIR = os.path.join(SCRIPT_PATH, "outputs/")
526
 
527
  # @spaces.GPU(duration=200)
528
  def tango(audio_path, character_name, seed=2024, create_graph=False, video_folder_path=None):
529
- download_files_from_repo()
530
  shutil.rmtree(OUTPUT_DIR, ignore_errors=True)
531
  os.makedirs(OUTPUT_DIR, exist_ok=True)
532
  cfg_file = os.path.join(SCRIPT_PATH, "configs/gradio.yaml")
 
1
+ from utils.download_utils import download_files_from_repo
2
+ download_files_from_repo()
3
+
4
  import os
5
  import gc
6
  import soundfile as sf
 
22
  import igraph
23
  import fire
24
 
 
25
  from utils.video_io import save_videos_from_pil
26
  from utils.genextend_inference_utils import adjust_statistics_to_match_reference
27
  from create_graph import path_visualization, graph_pruning, get_motion_reps_tensor, path_visualization_v2
 
528
 
529
  # @spaces.GPU(duration=200)
530
  def tango(audio_path, character_name, seed=2024, create_graph=False, video_folder_path=None):
 
531
  shutil.rmtree(OUTPUT_DIR, ignore_errors=True)
532
  os.makedirs(OUTPUT_DIR, exist_ok=True)
533
  cfg_file = os.path.join(SCRIPT_PATH, "configs/gradio.yaml")