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()