| 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, layout="vertical") | |
| download_button = gr.Button(download_model, "下载模型") | |
| iface.add_element(download_button) | |
| iface.launch() | |