fix: infer when no index file

app/infer.py

@@ -12,15 +12,19 @@ from model import device
 
 
 @zero(duration=120)
-def infer(exp_dir: str, original_audio: str, f0add: int) -> Tuple[int, np.ndarray]:
+def infer(
+    exp_dir: str, original_audio: str, f0add: int, index_rate: float, protect: float
+) -> Tuple[int, np.ndarray]:
     name = os.path.basename(exp_dir)
     model = os.path.join(exp_dir, f"{name}.pth")
     if not os.path.exists(model):
         raise gr.Error("Model not found")
 
     index = glob(f"{exp_dir}/added_*.index")
-    if not index:
-        raise gr.Error("Index not found")
+    if index:
+        index = index[0]
+    else:
+        index = None
 
     base = os.path.basename(original_audio)
     base = os.path.splitext(base)[0]
@@ -40,11 +44,11 @@ def infer(exp_dir: str, original_audio: str, f0add: int) -> Tuple[int, np.ndarra
         "rmvpe",
         index,
         None,
-        0.5,
-        3,
+        index_rate,
+        3,  # this only has effect when f0_method is "harvest"
         0,
         1,
-        0.33,
+        protect,
     )
 
     sr = wav_opt[0]
@@ -80,19 +84,36 @@ class InferenceTab:
         )
 
         with gr.Row():
-            self.original_audio = gr.Audio(
-                label="Upload original audio",
-                type="filepath",
-                show_download_button=True,
-            )
-            self.f0add = gr.Slider(
-                label="F0 add",
-                minimum=-16,
-                maximum=16,
-                step=1,
-                value=0,
-            )
-            self.infer_btn = gr.Button(value="Infer", variant="primary")
+            with gr.Column():
+                self.original_audio = gr.Audio(
+                    label="Upload original audio",
+                    type="filepath",
+                    show_download_button=True,
+                )
+            with gr.Column():
+                self.f0add = gr.Slider(
+                    label="F0 add",
+                    minimum=-16,
+                    maximum=16,
+                    step=1,
+                    value=0,
+                )
+                self.index_rate = gr.Slider(
+                    label="Index rate",
+                    minimum=-0,
+                    maximum=1,
+                    step=0.01,
+                    value=0.5,
+                )
+                self.protect = gr.Slider(
+                    label="Protect",
+                    minimum=0,
+                    maximum=1,
+                    step=0.01,
+                    value=0.33,
+                )
+            with gr.Column():
+                self.infer_btn = gr.Button(value="Infer", variant="primary")
         with gr.Row():
             self.infer_output = gr.Audio(label="Inferred audio")
         with gr.Row():
@@ -101,7 +122,13 @@ class InferenceTab:
     def build(self, exp_dir: gr.Textbox):
         self.infer_btn.click(
             fn=infer,
-            inputs=[exp_dir, self.original_audio, self.f0add],
+            inputs=[
+                exp_dir,
+                self.original_audio,
+                self.f0add,
+                self.index_rate,
+                self.protect,
+            ],
             outputs=[self.infer_output],
         ).success(
             fn=merge,