sdf / app.py
Virus561's picture
Update app.py
a12e909
import gradio as gr
import shutil
import subprocess
# Функция, которая будет вызываться при отправке данных
def save_file(input_file):
# Получаем путь к временному файлу
temp_file_path = input_file.name
# Указываем путь для сохранения файла
saved_file_path = "audio/saved_file.wav"
# Копируем временный файл в место сохранения
shutil.copy(temp_file_path, saved_file_path)
cmd = ["svc", "infer", f"/content/{saved_file_path}", "-m", "/content/models", "-c", "/content/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, # Отключаем режим реального времени
capture_session=True # Захватывать сессию для удаленного доступа
)
# Добавляем кнопку "Submit"
iface.launch(share=True)