Uniaff commited on
Commit
6c6b506
·
verified ·
1 Parent(s): c947273

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -20
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
- "--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
  ]
@@ -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=1 # Устанавливаем лимит на количество одновременно выполняемых операций
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
  # Запуск приложения