added assets

.gitattributes

@@ -39,3 +39,21 @@ pretrained_models/2stems/checkpoint filter=lfs diff=lfs merge=lfs -text
 pretrained_models/2stems/model.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text
 pretrained_models/2stems/model.index filter=lfs diff=lfs merge=lfs -text
 pretrained_models/2stems/model.meta filter=lfs diff=lfs merge=lfs -text
+assets/indices/carol_IVF59_Flat_nprobe_1_carol_v2.index filter=lfs diff=lfs merge=lfs -text
+assets/indices/me_IVF59_Flat_nprobe_1_me_v2.index filter=lfs diff=lfs merge=lfs -text
+assets/weights/francis.pth filter=lfs diff=lfs merge=lfs -text
+assets/weights/jess.pth filter=lfs diff=lfs merge=lfs -text
+assets/indices/peter_IVF39_Flat_nprobe_1_peter_v2.index filter=lfs diff=lfs merge=lfs -text
+assets/weights/daryl.pth filter=lfs diff=lfs merge=lfs -text
+assets/weights/elmo.pth filter=lfs diff=lfs merge=lfs -text
+assets/indices/daryl_IVF179_Flat_nprobe_1_daryl_v2.index filter=lfs diff=lfs merge=lfs -text
+assets/indices/elmomo_IVF59_Flat_nprobe_1_elmomo_v2.index filter=lfs diff=lfs merge=lfs -text
+assets/indices/francis_IVF13_Flat_nprobe_1_francis_v2.index filter=lfs diff=lfs merge=lfs -text
+assets/weights filter=lfs diff=lfs merge=lfs -text
+assets/weights/peter.pth filter=lfs diff=lfs merge=lfs -text
+assets/indices filter=lfs diff=lfs merge=lfs -text
+assets/indices/elmo_IVF129_Flat_nprobe_1_elmo_v2.index filter=lfs diff=lfs merge=lfs -text
+assets/indices/harry_IVF17_Flat_nprobe_1_harry_v2.index filter=lfs diff=lfs merge=lfs -text
+assets/indices/jess_IVF12_Flat_nprobe_1_jess_v2.index filter=lfs diff=lfs merge=lfs -text
+assets/weights/carol.pth filter=lfs diff=lfs merge=lfs -text
+assets/weights/harry.pth filter=lfs diff=lfs merge=lfs -text

app.py

@@ -7,13 +7,15 @@ import os
 from gradio_client import Client, file
 import subprocess
 
+from constant import weights, indices
+
 try:
   shutil.rmtree("output")
 except FileNotFoundError:
   pass
 
 
-def spleeter(aud, model_file, index_file, pitch):
+def spleeter(aud, model_dropdown, model_file, index_file, pitch):
   separator = Separator('spleeter:2stems')
   filename = os.path.basename(aud).split('.')[0]
   accompaniment_filename = f"./output/audio_example/{filename}_accompaniment.wav"
@@ -29,13 +31,17 @@ def spleeter(aud, model_file, index_file, pitch):
 
   # send vocal for processing
   # https://huggingface.co/spaces/r3gm/rvc_zero
+
+  model = model_file if model_dropdown is None else weights[model_dropdown]
+  index = index_file if model_dropdown is None else indices[model_dropdown]
+
   client = Client("r3gm/rvc_zero")
   result = client.predict(
     audio_files=[file(vocal_filename)],
-    file_m=file(model_file),
+    file_m=file(model),
     pitch_alg="rmvpe+",
     pitch_lvl=pitch,
-    file_index=file(index_file),
+    file_index=file(index),
     index_inf=0.75,
     r_m_f=3,
     e_r=0.25,
@@ -53,13 +59,14 @@ def spleeter(aud, model_file, index_file, pitch):
 
 inputs = [
           gr.Audio(sources='upload', label="Input Audio File", type="filepath"),
+    gr.Dropdown(choices=list(weights.keys()), value=list(weights.keys()), label="existing people model"),
 gr.File(
-        label="Model file",
+        label="Model file (use this if your existing people model does not exist in the dropdown)",
         type="filepath",
         height=130,
     ),
 gr.File(
-        label="Index file",
+        label="Index file (use this if your existing people model does not exist in the dropdown)",
         type="filepath",
         height=130,
     ),

assets/indices/carol_IVF59_Flat_nprobe_1_carol_v2.index

@@ -0,0 +1 @@
+logs/carol/added_IVF59_Flat_nprobe_1_carol_v2.index

assets/indices/daryl_IVF179_Flat_nprobe_1_daryl_v2.index

@@ -0,0 +1 @@
+logs/daryl/added_IVF179_Flat_nprobe_1_daryl_v2.index

assets/indices/elmo_IVF129_Flat_nprobe_1_elmo_v2.index

@@ -0,0 +1 @@
+logs/elmo/added_IVF129_Flat_nprobe_1_elmo_v2.index

assets/indices/francis_IVF13_Flat_nprobe_1_francis_v2.index

@@ -0,0 +1 @@
+logs/francis/added_IVF13_Flat_nprobe_1_francis_v2.index

assets/indices/harry_IVF17_Flat_nprobe_1_harry_v2.index

@@ -0,0 +1 @@
+logs/harry/added_IVF17_Flat_nprobe_1_harry_v2.index

assets/indices/jess_IVF12_Flat_nprobe_1_jess_v2.index

@@ -0,0 +1 @@
+logs/jess/added_IVF12_Flat_nprobe_1_jess_v2.index

assets/indices/peter_IVF39_Flat_nprobe_1_peter_v2.index

@@ -0,0 +1 @@
+logs/peter/added_IVF39_Flat_nprobe_1_peter_v2.index

assets/weights/carol.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2d66443b8733084d45bbf58b84eeeb4ad53ace2907c02a40aaa1a13b468822f2
+size 55187206

assets/weights/daryl.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ab05a09c230f190b149d3c71d24d9258becdee1079ce226e77daefe44ee8d243
+size 55210579

assets/weights/elmo.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d259715b2e8c46b8cc59349769e24856cfa57a8ad85ff296a6b741916d114ccb
+size 55186681

assets/weights/francis.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ad5b27c6ad3751299f8bbee318e0e1e10a3cbf8b08d66150a584c29b0bb5509c
+size 55216800

assets/weights/harry.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7e8c2df897b6181994e4289a027e3228790a827dcbfdc6a8fd803519670d1dbf
+size 55187206

assets/weights/jess.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c5a777b7d7d7a2abd23a938a164ce668c735110a33281519d949883dfceadbd0
+size 55186681

assets/weights/peter.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ab3f370dd46d0d90b7df4adb3791f035f43998f0b051b61e37c5f97e88eae793
+size 55187206

constant.py

@@ -0,0 +1,19 @@
+from glob import glob
+import os
+
+weight_files = glob('assets/weights/*', recursive=True)
+indices_files = glob('assets/indices/*', recursive=True)
+
+main_directory = "https://huggingface.co/spaces/darylfunggg/rvc-spleeter-voice/resolve/main/"
+
+weights = {'custom': None}
+for weight_file in weight_files:
+    weights[os.path.splitext(os.path.basename(weight_file))[0]] = f"{main_directory}{weight_file}"
+
+
+indices = {'custom': None}
+for indices_file in indices_files:
+    indices[os.path.basename(indices_file).split("_")[0]] = f"{main_directory}{indices_file}"
+
+
+assert len(weights) == len(indices)