Update app.py

app.py

@@ -5,6 +5,42 @@ from huggingface_hub import snapshot_download
 from ultimate_rvc.core.manage.config import load_config
 from ultimate_rvc.web.config.main import SettingsManagementConfig, TotalConfig
 
+embedders_list = [
+    ("embedders/contentvec/", ["pytorch_model.bin", "config.json"]),
+    ("embedders/custom/Crusty/", ["model.safetensors", "config.json"]),
+]
+
+class EmbedderModel(StrEnum):
+    """Enumeration of audio embedding models."""
+
+    CONTENTVEC = "contentvec"
+    CRUSTY = "Crusty"
+    CUSTOM = "custom"
+    
+class SeparationModel(StrEnum):
+    """Enumeration of audio separation models."""
+
+    UVR_MDX_NET_VOC_FT = "UVR-MDX-NET-Voc_FT.onnx"
+    UVR_MDX_NET_KARA_2 = "UVR_MDXNET_KARA_2.onnx"
+    REVERB_HQ_BY_FOXJOY = "Reverb_HQ_By_FoxJoy.onnx"
+    UVR-MDX-NET-Inst_HQ_3.onnx = "UVR-MDX-NET-Inst_HQ_3.onnx"
+    UVR-MDX-NET-Voc_FT.onnx = "UVR-MDX-NET-Voc_FT.onnx"
+    Kim_Vocal_1.onnx = "Kim_Vocal_1.onnx"
+    Kim_Vocal_2.onnx = "Kim_Vocal_2.onnx"
+    Kim_Inst.onnx = "Kim_Inst.onnx"
+    UVR-MDX-NET-Inst_HQ_4.onnx = "UVR-MDX-NET-Inst_HQ_4.onnx"
+    kuielab_a_vocals.onnx = "kuielab_a_vocals.onnx"
+    kuielab_b_vocals.onnx = "kuielab_b_vocals.onnx"
+    kuielab_a_drums.onnx = "kuielab_a_drums.onnx"
+    kuielab_b_drums.onnx = "kuielab_b_drums.onnx"
+    kuielab_a_bass.onnx = "kuielab_a_bass.onnx"
+    kuielab_b_bass.onnx = "kuielab_b_bass.onnx"
+    kuielab_a_other.onnx = "kuielab_a_other.onnx"
+    kuielab_b_other.onnx = "kuielab_b_other.onnx"
+    MDX23C-8KFFT-InstVoc_HQ_2.ckpt = "MDX23C-8KFFT-InstVoc_HQ_2.ckpt"
+    UVR-DeNoise.pth = "UVR-DeNoise.pth"
+
+
 load_config("default", TotalConfig)
 
 now_dir = os.getcwd()