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 |
# Запуск приложения
|