Update app.py

app.py

@@ -1,24 +1,21 @@
 import gradio as gr
-import subprocess
-import os
+import requests
 
-def download_model():
-    # 创建目录 ./trained/wdlm/，如果不存在的话
-    subprocess.run(["mkdir", "-p", "./trained/wdlm/"])
+def download_file(download_url, save_path):
+    try:
+        response = requests.get(download_url)
+        with open(save_path, 'wb') as f:
+            f.write(response.content)
+        return f"File downloaded successfully to {save_path}"
+    except Exception as e:
+        return f"Error downloading file: {str(e)}"
 
-    # 切换到目录 ./trained/wdlm/
-    os.chdir("./trained/wdlm/")
+iface = gr.Interface(fn=download_file, inputs=["text", "text"], outputs="text",
+                     inputs_label=["Download URL", "Save Path"],
+                     outputs_label="Download Status")
+iface.launch(share=True)
 
-    # 下载模型文件
-    subprocess.run(["wget", "-O", "G_38400.pth", "https://github.com/00000000002/render/releases/download/ddd/G_38400.pth"])
+# 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()