app.py

import gradio as gr
import subprocess
import os

def download_model():
    # 创建目录 ./trained/wdlm/，如果不存在的话
    subprocess.run(["mkdir", "-p", "./trained/wdlm/"])

    # 切换到目录 ./trained/wdlm/
    os.chdir("./trained/wdlm/")

    # 下载模型文件
    subprocess.run(["wget", "-O", "G_38400.pth", "https://github.com/00000000002/render/releases/download/ddd/G_38400.pth"])

    print("模型下载完成！")

iface = gr.Interface(fn=download_model, inputs="None", outputs="None", live=True)
download_button = gr.Button(download_model, "下载模型")
download_button.style.justify_content = "center"  # 垂直居中
download_button.style.width = "200px"  # 设置按钮宽度
iface.add_element(download_button)
iface.launch()