Uniaff commited on
Commit
767269a
·
verified ·
1 Parent(s): 95e31cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -10
app.py CHANGED
@@ -10,21 +10,22 @@ def generate(video, audio, checkpoint, no_smooth, resize_factor, pad_top, pad_bo
10
  if video is None or audio is None or checkpoint is None:
11
  return "Пожалуйста, загрузите видео/изображение и аудио файл, а также выберите чекпойнт."
12
 
13
- # Поскольку Gradio теперь возвращает пути к файлам, мы используем их напрямую без необходимости копировать их содержимое
14
- video_path = video # Уже строка с путем к видео
15
- audio_path = audio # Уже строка с путем к аудио
16
 
17
- # Создание временной директории для сохранения выходного видео
18
  with tempfile.TemporaryDirectory() as temp_dir:
19
  output_path = os.path.join(temp_dir, "output.mp4")
20
 
 
 
 
21
  # Подготовка аргументов для инференса
22
  args = [
23
  "--checkpoint_path", f"checkpoints/{checkpoint}.pth",
24
  "--segmentation_path", "checkpoints/face_segmentation.pth",
25
  "--enhance_face", "gfpgan",
26
- "--face", video_path,
27
- "--audio", audio_path,
28
  "--outfile", output_path,
29
  "--resize_factor", str(resize_factor),
30
  "--pads", str(pad_top), str(pad_bottom), str(pad_left), str(pad_right)
@@ -34,18 +35,19 @@ def generate(video, audio, checkpoint, no_smooth, resize_factor, pad_top, pad_bo
34
  args.append("--nosmooth")
35
 
36
  try:
37
- # Вызов функции инференса через subprocess
 
 
38
  cmd = ["python", "inference.py"] + args
39
  subprocess.run(cmd, check=True)
40
  except subprocess.CalledProcessError as e:
41
  return f"Произошла ошибка при обработке: {e}"
42
 
43
- # Проверка наличия выходного файла
44
  if not os.path.exists(output_path):
45
  return "Не удалось создать выходное видео."
46
 
47
- # Возвращаем путь к выходному файлу
48
- return output_path
49
 
50
  with gr.Blocks() as ui:
51
  gr.Markdown("## Wav2Lip - Синхронизация губ в видео")
 
10
  if video is None or audio is None or checkpoint is None:
11
  return "Пожалуйста, загрузите видео/изображение и аудио файл, а также выберите чекпойнт."
12
 
13
+ # Лог для видео и аудио путей
14
+ print(f"Video path: {video}, Audio path: {audio}")
 
15
 
 
16
  with tempfile.TemporaryDirectory() as temp_dir:
17
  output_path = os.path.join(temp_dir, "output.mp4")
18
 
19
+ # Лог для вывода пути к файлу
20
+ print(f"Output path: {output_path}")
21
+
22
  # Подготовка аргументов для инференса
23
  args = [
24
  "--checkpoint_path", f"checkpoints/{checkpoint}.pth",
25
  "--segmentation_path", "checkpoints/face_segmentation.pth",
26
  "--enhance_face", "gfpgan",
27
+ "--face", video,
28
+ "--audio", audio,
29
  "--outfile", output_path,
30
  "--resize_factor", str(resize_factor),
31
  "--pads", str(pad_top), str(pad_bottom), str(pad_left), str(pad_right)
 
35
  args.append("--nosmooth")
36
 
37
  try:
38
+ # Лог для команды инференса
39
+ print(f"Running inference with command: python inference.py {' '.join(args)}")
40
+
41
  cmd = ["python", "inference.py"] + args
42
  subprocess.run(cmd, check=True)
43
  except subprocess.CalledProcessError as e:
44
  return f"Произошла ошибка при обработке: {e}"
45
 
 
46
  if not os.path.exists(output_path):
47
  return "Не удалось создать выходное видео."
48
 
49
+ return output_path
50
+
51
 
52
  with gr.Blocks() as ui:
53
  gr.Markdown("## Wav2Lip - Синхронизация губ в видео")