resilient to invalid index files

src/main.py

@@ -100,12 +100,17 @@ def get_rvc_model(voice_model, is_webui):
     for root, dirs, files in os.walk(model_dir):
         # Iterate over the files in the current directory
         for file in files:
+            # Check if the file has a .pth extension
             if file.endswith('.pth'):
                 # Append the full path of the file to the list
                 rvc_model_filename = os.path.join(root, file)
-            #if file.endswith('.index'):
-            #    rvc_index_filename = file
-
+            if file.endswith('.index'):
+                filename = os.path.join(root, file)
+                if rvc_index_filename != None:
+                    if os.path.getsize(filename) >os.path.getsize(rvc_index_filename):
+                        rvc_index_filename = filename
+                else:
+                    rvc_index_filename = filename
     if rvc_model_filename is None:
         error_msg = f'No model file exists in {model_dir}.'
         raise_exception(error_msg, is_webui)

src/vc_infer_pipeline.py

@@ -365,7 +365,7 @@ class VC(object):
         ) + 1
         f0_mel[f0_mel <= 1] = 1
         f0_mel[f0_mel > 255] = 255
-        f0_coarse = np.rint(f0_mel).astype(np.int)
+        f0_coarse = np.rint(f0_mel).astype(int)
 
         return f0_coarse, f0bak  # 1-0
 
@@ -504,7 +504,10 @@ class VC(object):
             try:
                 index = faiss.read_index(file_index)
                 # big_npy = np.load(file_big_npy)
-                big_npy = index.reconstruct_n(0, index.ntotal)
+                if index.ntotal == 0:
+                    index = big_npy = None
+                else:
+                    big_npy = index.reconstruct_n(0, index.ntotal)
             except:
                 traceback.print_exc()
                 index = big_npy = None