Spaces:
Running
Running
Update app.py
Browse files
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 |
-
# Вызов команды
|
40 |
cmd = ["python", "inference.py"] + args
|
41 |
print(f"Running inference with command: {' '.join(cmd)}")
|
42 |
-
subprocess.run(cmd, check=True
|
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,
|
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("Сгенерировать")
|