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}"
|