Uniaff commited on
Commit
c947273
·
verified ·
1 Parent(s): 7ad861e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +25 -8
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
- "--checkpoint_path", f"checkpoints/{checkpoint}.pth",
33
- "--segmentation_path", "checkpoints/face_segmentation.pth",
34
- "--enhance_face", "gfpgan",
35
- "--face", video_path,
36
- "--audio", audio_path,
37
- "--outfile", output_path,
38
- "--resize_factor", str(resize_factor),
39
- "--pads", str(pad_top), str(pad_bottom), str(pad_left), str(pad_right)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: