Update app.py

app.py

@@ -1,14 +1,20 @@
 import gradio as gr
 import subprocess
+import os
 
-def greet(name):
+def download_model():
+    # 创建目录 ./trained/wdlm/，如果不存在的话
     subprocess.run(["mkdir", "-p", "./trained/wdlm/"])
-        
-        # 切换到目录 ./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"])
 
-    return "Hello " + name + "!!"
+    print("模型下载完成！")
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+iface = gr.Interface(fn=lambda: None, live=True, layout="vertical")
+download_button = gr.Button(download_model, "下载模型")
+iface.add_element(download_button)
+iface.launch()