trying to fix for multiple models

app.py

@@ -62,12 +62,24 @@ else:
     cluster_model_path = None
 default_cluster_infer_ratio = default_cluster_infer_ratio if cluster_model_path else 0
 
-generator_path = hf_hub_download(repo_id, ckpt_name)
+#generator_path = hf_hub_download(repo_id, ckpt_name)
+generator_path = None
+if speaker == speakers[0]:
+    generator_path = hf_hub_download(repo_id, "G_10000.pth")
+elif speaker == speakers[1]:
+    generator_path = hf_hub_download(repo_id, "G_534.pth")
+elif speaker == speakers[2]:
+    generator_path = hf_hub_download(repo_id, "G_9933.pth")
+else:
+    # Handle the case when the speaker type is not recognized
+    raise ValueError("Invalid speaker type")
+
+model = Svc(net_g_path=generator_path, config_path=config_path, device=device, cluster_model_path=cluster_model_path)
 config_path = hf_hub_download(repo_id, "config.json")
 hparams = HParams(**json.loads(Path(config_path).read_text()))
 speakers = list(hparams.spk.keys())
 device = "cuda" if torch.cuda.is_available() else "cpu"
-model = Svc(net_g_path=generator_path, config_path=config_path, device=device, cluster_model_path=cluster_model_path)
+#model = Svc(net_g_path=generator_path, config_path=config_path, device=device, cluster_model_path=cluster_model_path)
 demucs_model = get_model(DEFAULT_MODEL)