Spaces:
Running
Running
import gradio as gr | |
import os | |
import subprocess, os | |
assets_folder = "assets" | |
if not os.path.exists(assets_folder): | |
os.makedirs(assets_folder) | |
files = { | |
"rmvpe/rmvpe.pt":"https://huggingface.co/Rejekts/project/resolve/main/rmvpe.pt", | |
"hubert/hubert_base.pt":"https://huggingface.co/Rejekts/project/resolve/main/hubert_base.pt", | |
"pretrained_v2/D40k.pth":"https://huggingface.co/Rejekts/project/resolve/main/D40k.pth", | |
"pretrained_v2/G40k.pth":"https://huggingface.co/Rejekts/project/resolve/main/G40k.pth", | |
"pretrained_v2/f0D40k.pth":"https://huggingface.co/Rejekts/project/resolve/main/f0D40k.pth", | |
"pretrained_v2/f0G40k.pth":"https://huggingface.co/Rejekts/project/resolve/main/f0G40k.pth" | |
} | |
for file, link in files.items(): | |
file_path = os.path.join(assets_folder, file) | |
if not os.path.exists(file_path): | |
try: | |
subprocess.run(['wget', link, '-O', file_path], check=True) | |
except subprocess.CalledProcessError as e: | |
print(f"Error downloading {file}: {e}") | |
def sorted(filepath): | |
return os.listdir(filepath) | |
with gr.Blocks(title="🔊",theme=gr.themes.Base(primary_hue="rose",neutral_hue="zinc")) as app: | |
with gr.Row(): | |
weight = gr.Dropdown(choices=sorted('assets/weights')) | |
app.launch() |