Update app.py

app.py

@@ -1,9 +1,10 @@
 from web import *
 import gradio as gr
 import os
-
-
-import os
+import shutil, glob
+from easyfuncs import download_from_url, CachedModels
+os.makedirs("dataset",exist_ok=True)
+model_library = CachedModels()
 
 assets_folder = "./assets/"
 if not os.path.exists(assets_folder):
@@ -71,7 +72,24 @@ with gr.Blocks(title="RVC UI") as app:
                         label="Output information",
                         interactive=False
                     )
-    
+    with gr.TabItem("Download Models"):
+        with gr.Row():
+            url_input = gr.Textbox(label="URL to model", value="",placeholder="https://...", scale=6)
+            name_output = gr.Textbox(label="Save as", value="",placeholder="MyModel",scale=2)
+            url_download = gr.Button(value="Download Model",scale=2)
+            url_download.click(
+                inputs=[url_input,name_output],
+                outputs=[url_input],
+                fn=download_from_url,
+            )
+        with gr.Row():
+            model_browser = gr.Dropdown(choices=list(model_library.models.keys()),label="OR Search Models (Quality UNKNOWN)",scale=5)
+            download_from_browser = gr.Button(value="Get",scale=2)
+            download_from_browser.click(
+                inputs=[model_browser],
+                outputs=[model_browser],
+                fn=lambda model: download_from_url(model_library.models[model],model),
+            )
     with gr.TabItem("Train"):
         gr.Markdown("### Step 1. Fill in the experimental configuration.\nExperimental data is stored in the 'logs' folder, with each experiment having a separate folder. Manually enter the experiment name path, which contains the experimental configuration, logs, and trained model files.")
         exp_dir1 = gr.Textbox(label="Enter the experiment name", value="my voice")