Update src/main.py

src/main.py

@@ -100,10 +100,11 @@ 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 rvc_model_filename is None:
         error_msg = f'No model file exists in {model_dir}.'