add

app.py

@@ -54,17 +54,20 @@ for folder_name in folder_names:
     models_info[speaker]["config"] = f"{config_path}/config.json"
     if os.path.exists(f"{models_path}/feature_and_index.pkl"):
         models_info[speaker]["cluster"] = f"{models_path}/feature_and_index.pkl"
+        models_info[speaker]["feature_retrieval"] = True
     elif os.path.exists(f"{models_path}/kmeans_10000.pt"):
         models_info[speaker]["cluster"] = f"{models_path}/kmeans_10000.pt"
+        models_info[speaker]["feature_retrieval"] = False
     else:
         models_info[speaker]["cluster"] = "logs/44k/kmeans_10000.pt"
+        models_info[speaker]["feature_retrieval"] = False
 speakers = list(models_info.keys())
 
 def load(speaker):
     global sid
     global model
     sid = speaker
-    model = Svc(models_info[speaker]["model"], models_info[speaker]["config"], cluster_model_path=models_info[speaker]["cluster"])
+    model = Svc(models_info[speaker]["model"], models_info[speaker]["config"], cluster_model_path=models_info[speaker]["cluster"], feature_retrieval=models_info[speaker]["feature_retrieval"])
     return "加载成功"
 load(speakers[0])