Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -10,21 +10,22 @@ def generate(video, audio, checkpoint, no_smooth, resize_factor, pad_top, pad_bo
|
|
10 |
if video is None or audio is None or checkpoint is None:
|
11 |
return "Пожалуйста, загрузите видео/изображение и аудио файл, а также выберите чекпойнт."
|
12 |
|
13 |
-
#
|
14 |
-
|
15 |
-
audio_path = audio # Уже строка с путем к аудио
|
16 |
|
17 |
-
# Создание временной директории для сохранения выходного видео
|
18 |
with tempfile.TemporaryDirectory() as temp_dir:
|
19 |
output_path = os.path.join(temp_dir, "output.mp4")
|
20 |
|
|
|
|
|
|
|
21 |
# Подготовка аргументов для инференса
|
22 |
args = [
|
23 |
"--checkpoint_path", f"checkpoints/{checkpoint}.pth",
|
24 |
"--segmentation_path", "checkpoints/face_segmentation.pth",
|
25 |
"--enhance_face", "gfpgan",
|
26 |
-
"--face",
|
27 |
-
"--audio",
|
28 |
"--outfile", output_path,
|
29 |
"--resize_factor", str(resize_factor),
|
30 |
"--pads", str(pad_top), str(pad_bottom), str(pad_left), str(pad_right)
|
@@ -34,18 +35,19 @@ def generate(video, audio, checkpoint, no_smooth, resize_factor, pad_top, pad_bo
|
|
34 |
args.append("--nosmooth")
|
35 |
|
36 |
try:
|
37 |
-
#
|
|
|
|
|
38 |
cmd = ["python", "inference.py"] + args
|
39 |
subprocess.run(cmd, check=True)
|
40 |
except subprocess.CalledProcessError as e:
|
41 |
return f"Произошла ошибка при обработке: {e}"
|
42 |
|
43 |
-
# Проверка наличия выходного файла
|
44 |
if not os.path.exists(output_path):
|
45 |
return "Не удалось создать выходное видео."
|
46 |
|
47 |
-
|
48 |
-
|
49 |
|
50 |
with gr.Blocks() as ui:
|
51 |
gr.Markdown("## Wav2Lip - Синхронизация губ в видео")
|
|
|
10 |
if video is None or audio is None or checkpoint is None:
|
11 |
return "Пожалуйста, загрузите видео/изображение и аудио файл, а также выберите чекпойнт."
|
12 |
|
13 |
+
# Лог для видео и аудио путей
|
14 |
+
print(f"Video path: {video}, Audio path: {audio}")
|
|
|
15 |
|
|
|
16 |
with tempfile.TemporaryDirectory() as temp_dir:
|
17 |
output_path = os.path.join(temp_dir, "output.mp4")
|
18 |
|
19 |
+
# Лог для вывода пути к файлу
|
20 |
+
print(f"Output path: {output_path}")
|
21 |
+
|
22 |
# Подготовка аргументов для инференса
|
23 |
args = [
|
24 |
"--checkpoint_path", f"checkpoints/{checkpoint}.pth",
|
25 |
"--segmentation_path", "checkpoints/face_segmentation.pth",
|
26 |
"--enhance_face", "gfpgan",
|
27 |
+
"--face", video,
|
28 |
+
"--audio", audio,
|
29 |
"--outfile", output_path,
|
30 |
"--resize_factor", str(resize_factor),
|
31 |
"--pads", str(pad_top), str(pad_bottom), str(pad_left), str(pad_right)
|
|
|
35 |
args.append("--nosmooth")
|
36 |
|
37 |
try:
|
38 |
+
# Лог для команды инференса
|
39 |
+
print(f"Running inference with command: python inference.py {' '.join(args)}")
|
40 |
+
|
41 |
cmd = ["python", "inference.py"] + args
|
42 |
subprocess.run(cmd, check=True)
|
43 |
except subprocess.CalledProcessError as e:
|
44 |
return f"Произошла ошибка при обработке: {e}"
|
45 |
|
|
|
46 |
if not os.path.exists(output_path):
|
47 |
return "Не удалось создать выходное видео."
|
48 |
|
49 |
+
return output_path
|
50 |
+
|
51 |
|
52 |
with gr.Blocks() as ui:
|
53 |
gr.Markdown("## Wav2Lip - Синхронизация губ в видео")
|