Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -29,14 +29,31 @@ def generate(video, audio, checkpoint, no_smooth, resize_factor, pad_top, pad_bo
|
|
29 |
|
30 |
# Подготовка аргументов для инференса
|
31 |
args = [
|
32 |
-
|
33 |
-
|
34 |
-
|
35 |
-
|
36 |
-
|
37 |
-
|
38 |
-
|
39 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
40 |
]
|
41 |
|
42 |
if no_smooth:
|
|
|
29 |
|
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 |
+
"--resize_factor", "2", # Установите 2 или больше для ускорения
|
48 |
+
|
49 |
+
# Увеличение размеров батча для более эффективного использования GPU
|
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 |
]
|
58 |
|
59 |
if no_smooth:
|