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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
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}"