Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -30,28 +30,17 @@ def generate(video, audio, checkpoint, no_smooth, resize_factor, pad_top, pad_bo
|
|
30 |
# Подготовка аргументов для инференса
|
31 |
args = [
|
32 |
"--checkpoint_path", f"checkpoints/{checkpoint}.pth",
|
33 |
-
"--segmentation_path", "checkpoints/face_segmentation.pth",
|
34 |
-
|
35 |
-
|
36 |
-
# Если улучшение лица необходимо, оставьте эту строку
|
37 |
-
# "--enhance_face", "gfpgan",
|
38 |
-
|
39 |
-
"--no_seg", # Отключение сегментации лица
|
40 |
-
"--no_sr", # Отключение суперразрешения
|
41 |
-
|
42 |
"--face", video_path,
|
43 |
"--audio", audio_path,
|
44 |
"--outfile", output_path,
|
45 |
-
|
46 |
-
|
47 |
-
"--
|
48 |
-
|
49 |
-
|
50 |
-
"--face_det_batch_size", "32", # Повышение размера батча для обнаружения лиц
|
51 |
-
"--wav2lip_batch_size", "256", # Повышение размера батча для Wav2Lip модели
|
52 |
-
|
53 |
-
"--static", # Использование только первого кадра для инференса
|
54 |
-
"--fps", "15", # Уменьшение FPS для статичного изображения
|
55 |
|
56 |
"--pads", str(pad_top), str(pad_bottom), str(pad_left), str(pad_right)
|
57 |
]
|
@@ -106,7 +95,7 @@ with gr.Blocks() as ui:
|
|
106 |
generate,
|
107 |
inputs=[video, audio, checkpoint, no_smooth, resize_factor, pad_top, pad_bottom, pad_left, pad_right, save_as_video],
|
108 |
outputs=result,
|
109 |
-
concurrency_limit=
|
110 |
)
|
111 |
|
112 |
# Запуск приложения
|
|
|
30 |
# Подготовка аргументов для инференса
|
31 |
args = [
|
32 |
"--checkpoint_path", f"checkpoints/{checkpoint}.pth",
|
33 |
+
"--segmentation_path", "checkpoints/face_segmentation.pth",
|
34 |
+
"--no_seg",
|
35 |
+
"--no_sr",
|
|
|
|
|
|
|
|
|
|
|
|
|
36 |
"--face", video_path,
|
37 |
"--audio", audio_path,
|
38 |
"--outfile", output_path,
|
39 |
+
"--resize_factor", "4",
|
40 |
+
"--face_det_batch_size", "32",
|
41 |
+
"--wav2lip_batch_size", "256",
|
42 |
+
"--static",
|
43 |
+
"--fps", "15",
|
|
|
|
|
|
|
|
|
|
|
44 |
|
45 |
"--pads", str(pad_top), str(pad_bottom), str(pad_left), str(pad_right)
|
46 |
]
|
|
|
95 |
generate,
|
96 |
inputs=[video, audio, checkpoint, no_smooth, resize_factor, pad_top, pad_bottom, pad_left, pad_right, save_as_video],
|
97 |
outputs=result,
|
98 |
+
concurrency_limit=10 # Устанавливаем лимит на количество одновременно выполняемых операций
|
99 |
)
|
100 |
|
101 |
# Запуск приложения
|