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:
|