Update app.py

app.py

@@ -15,7 +15,6 @@ from config import is_half, device
 
 logging.getLogger("numba").setLevel(logging.WARNING)
 
-
 def create_vc_fn(tgt_sr, net_g, vc, if_f0, file_index, file_big_npy):
     def vc_fn(input_audio, f0_up_key, f0_method, index_rate):
         try:
@@ -59,7 +58,6 @@ def create_vc_fn(tgt_sr, net_g, vc, if_f0, file_index, file_big_npy):
 
     return vc_fn
 
-
 def load_hubert():
     global hubert_model
     models, _, _ = checkpoint_utils.load_model_ensemble_and_task(
@@ -74,17 +72,19 @@ def load_hubert():
         hubert_model = hubert_model.float()
     hubert_model.eval()
 
-
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()
     parser.add_argument('--api', action="store_true", default=False)
     parser.add_argument("--share", action="store_true", default=False, help="share gradio app")
     parser.add_argument("--files", action="store_true", default=False, help="load audio from path")
     args, unknown = parser.parse_known_args()
+    
     load_hubert()
     models = []
+    
     with open("weights/model_info.json", "r", encoding="utf-8") as f:
         models_info = json.load(f)
+    
     for name, info in models_info.items():
         if not info['enable']:
             continue
@@ -95,6 +95,7 @@ if __name__ == '__main__':
         cpt = torch.load(f"weights/{name}/{name}.pth", map_location="cpu")
         tgt_sr = cpt["config"][-1]
         if_f0 = cpt.get("f0", 1)
+        
         if if_f0 == 1:
             net_g = SynthesizerTrnMs256NSFsid(*cpt["config"], is_half=is_half)
         else:
@@ -102,10 +103,12 @@ if __name__ == '__main__':
         del net_g.enc_q
         print(net_g.load_state_dict(cpt["weight"], strict=False))
         net_g.eval().to(device)
+        
         if is_half:
             net_g = net_g.half()
         else:
             net_g = net_g.float()
+        
         vc = VC(tgt_sr, device, is_half)
         models.append((name, title, cover, create_vc_fn(tgt_sr, net_g, vc, if_f0, index, npy)))
 
@@ -152,16 +155,16 @@ if __name__ == '__main__':
                 vc_submit.click(vc_fn, [vc_input, vc_transpose, vc_f0method, vc_index_ratio], [vc_output1, vc_output2])
         app.queue(concurrency_count=1, max_size=20, api_open=args.api).launch(share=args.share)
 
-# Note: The following code is outside the `__main__` block, as it was causing indentation issues in the provided code.
-iface = gr.Interface(
-    fn=create_vc_fn(tgt_sr, net_g, vc, if_f0, file_index, file_big_npy),
-    inputs=[
-        gr.inputs.Audio(source="microphone", type="numpy", sample_rate=16000),
-        gr.inputs.Slider(minimum=-12, maximum=12, step=1, default=0),
-        gr.inputs.Radio(["world", "dio"], label="F0 method"),
-        gr.inputs.Slider(minimum=0, maximum=1, step=0.01, default=0.5)
-    ],
-    outputs="text",
-)
+    # Note: The following code is outside the `__main__` block, as it was causing indentation issues in the provided code.
+    iface = gr.Interface(
+        fn=create_vc_fn(tgt_sr, net_g, vc, if_f0, file_index, file_big_npy),
+        inputs=[
+            gr.inputs.Audio(source="microphone", type="numpy", sample_rate=16000),
+            gr.inputs.Slider(minimum=-12, maximum=12, step=1, default=0),
+            gr.inputs.Radio(["world", "dio"], label="F0 method"),
+            gr.inputs.Slider(minimum=0, maximum=1, step=0.01, default=0.5)
+        ],
+        outputs="text",
+    )
 
-iface.launch()
+    iface.launch()