Uniaff commited on
Commit
c24d132
·
verified ·
1 Parent(s): 11cf479

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -7
app.py CHANGED
@@ -3,6 +3,9 @@ import subprocess
3
  import os
4
  import tempfile
5
 
 
 
 
6
  def generate(video, audio, checkpoint, no_smooth, resize_factor, pad_top, pad_bottom, pad_left, pad_right):
7
  if video is None or audio is None or checkpoint is None:
8
  return "Пожалуйста, загрузите видео/изображение и аудио файл, а также выберите чекпойнт."
@@ -13,10 +16,6 @@ def generate(video, audio, checkpoint, no_smooth, resize_factor, pad_top, pad_bo
13
 
14
  # Создание временной директории для сохранения выходного видео
15
  with tempfile.TemporaryDirectory() as temp_dir:
16
- # Создание поддиректории 'temp' внутри temp_dir для inference.py
17
- temp_subdir = os.path.join(temp_dir, 'temp')
18
- os.makedirs(temp_subdir, exist_ok=True)
19
-
20
  # Определение выходного файла
21
  output_path = os.path.join(temp_dir, "output.mp4")
22
 
@@ -36,10 +35,10 @@ def generate(video, audio, checkpoint, no_smooth, resize_factor, pad_top, pad_bo
36
  args.append("--nosmooth")
37
 
38
  try:
39
- # Вызов команды с установкой рабочей директории в temp_dir
40
  cmd = ["python", "inference.py"] + args
41
  print(f"Running inference with command: {' '.join(cmd)}")
42
- subprocess.run(cmd, check=True, cwd=temp_dir)
43
  except subprocess.CalledProcessError as e:
44
  return f"Произошла ошибка при обработке: {e}"
45
 
@@ -62,7 +61,7 @@ with gr.Blocks() as ui:
62
  with gr.Row():
63
  with gr.Column():
64
  pad_top = gr.Slider(minimum=0, maximum=50, step=1, value=0, label="Отступ сверху")
65
- pad_bottom = gr.Slider(minimum=0, maximum=50, step=1, value=10, label="Отступ снизу (рекомендуется 20 для включения подбородка)")
66
  pad_left = gr.Slider(minimum=0, maximum=50, step=1, value=0, label="Отступ слева")
67
  pad_right = gr.Slider(minimum=0, maximum=50, step=1, value=0, label="Отступ справа")
68
  generate_btn = gr.Button("Сгенерировать")
 
3
  import os
4
  import tempfile
5
 
6
+ print(os.listdir())
7
+ print(os.getcwd())
8
+
9
  def generate(video, audio, checkpoint, no_smooth, resize_factor, pad_top, pad_bottom, pad_left, pad_right):
10
  if video is None or audio is None or checkpoint is None:
11
  return "Пожалуйста, загрузите видео/изображение и аудио файл, а также выберите чекпойнт."
 
16
 
17
  # Создание временной директории для сохранения выходного видео
18
  with tempfile.TemporaryDirectory() as temp_dir:
 
 
 
 
19
  # Определение выходного файла
20
  output_path = os.path.join(temp_dir, "output.mp4")
21
 
 
35
  args.append("--nosmooth")
36
 
37
  try:
38
+ # Вызов команды без изменения рабочей директории
39
  cmd = ["python", "inference.py"] + args
40
  print(f"Running inference with command: {' '.join(cmd)}")
41
+ subprocess.run(cmd, check=True) # Убрали cwd=temp_dir, чтобы использовать текущую директорию
42
  except subprocess.CalledProcessError as e:
43
  return f"Произошла ошибка при обработке: {e}"
44
 
 
61
  with gr.Row():
62
  with gr.Column():
63
  pad_top = gr.Slider(minimum=0, maximum=50, step=1, value=0, label="Отступ сверху")
64
+ pad_bottom = gr.Slider(minimum=0, максимум 50, step=1, value=10, label="Отступ снизу (рекомендуется 20 для включения подбородка)")
65
  pad_left = gr.Slider(minimum=0, maximum=50, step=1, value=0, label="Отступ слева")
66
  pad_right = gr.Slider(minimum=0, maximum=50, step=1, value=0, label="Отступ справа")
67
  generate_btn = gr.Button("Сгенерировать")