zishuo/TANGO
1
0
Fork 0
mirror of https://huggingface.co/spaces/H-Liu1997/TANGO synced 2025-12-24 13:29:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update inference.py

Browse Source
This commit is contained in:
Haiyang Liu
2025-06-24 15:40:42 +00:00
committed by system
parent 534c45b130
commit dac4f1f3ba
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
inference.py
View File

@@ -1,3 +1,6 @@
from utils.download_utils import download_files_from_repo
download_files_from_repo()
import os
import gc
import soundfile as sf
@@ -19,7 +22,6 @@ import smplx
import igraph
import fire
from utils.download_utils import download_files_from_repo
from utils.video_io import save_videos_from_pil
from utils.genextend_inference_utils import adjust_statistics_to_match_reference
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/")
# @spaces.GPU(duration=200)
def tango(audio_path, character_name, seed=2024, create_graph=False, video_folder_path=None):
download_files_from_repo()
shutil.rmtree(OUTPUT_DIR, ignore_errors=True)
os.makedirs(OUTPUT_DIR, exist_ok=True)
cfg_file = os.path.join(SCRIPT_PATH, "configs/gradio.yaml")