egegdvd / app.py
Virus561's picture
Update app.py
666e618
raw
history blame
No virus
1.43 kB
import gradio as gr
import subprocess
# Функция, которая будет вызываться при отправке данных
def save_file(input_file):
# Указываем путь для сохранения файла
saved_file_path = "/home/user/app/audio/saved_file.wav" # Укажите полный путь до целевого файла
# Сохраняем содержимое input_file в целевой файл
with open(saved_file_path, 'wb') as saved_file:
saved_file.write(input_file.read())
cmd = ["svc", "infer", saved_file_path, "-m", "/home/user/app/models", "-c", "/home/user/app/models/config.json"]
result = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
print("STDOUT:", result.stdout)
print("STDERR:", result.stderr)
# Получаем байтовый код финального файла
with open(saved_file_path, "rb") as f:
final_file_bytes = f.read()
return final_file_bytes
# Создание интерфейса Gradio с кнопкой "Submit"
iface = gr.Interface(
fn=save_file,
inputs="file", # Тип ввода - файл
outputs="text", # Тип вывода - текстовая строка
live=False, # Отключаем режим реального времени
)
# Добавляем кнопку "Submit"
iface.launch()