Update inference.py
Browse files- 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")
|