Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -1,10 +1,7 @@
|
|
1 |
import gradio as gr
|
2 |
-
import torch
|
3 |
-
import os
|
4 |
import subprocess
|
|
|
5 |
import tempfile
|
6 |
-
from inference import main as inference_main # Предполагается, что вы интегрируете функцию main из inference.py
|
7 |
-
import shutil
|
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:
|
@@ -41,6 +38,7 @@ def generate(video, audio, checkpoint, no_smooth, resize_factor, pad_top, pad_bo
|
|
41 |
try:
|
42 |
# Вызов команды с установкой рабочей директории в temp_dir
|
43 |
cmd = ["python", "inference.py"] + args
|
|
|
44 |
subprocess.run(cmd, check=True, cwd=temp_dir)
|
45 |
except subprocess.CalledProcessError as e:
|
46 |
return f"Произошла ошибка при обработке: {e}"
|
|
|
1 |
import gradio as gr
|
|
|
|
|
2 |
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:
|
|
|
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}"
|