diff --git a/tabs/inference/inference.py b/tabs/inference/inference.py index bced5ffc..f6e03800 100644 --- a/tabs/inference/inference.py +++ b/tabs/inference/inference.py @@ -322,7 +322,7 @@ def refresh_embedders_folders(): def get_speakers_id(model): if model: - model_data = torch.load(os.path.join(now_dir,model), map_location="cpu") + model_data = torch.load(os.path.join(now_dir, model), map_location="cpu") speakers_id = model_data.get("speakers_id") if speakers_id: return list(range(speakers_id)) @@ -331,6 +331,7 @@ def get_speakers_id(model): else: return [0] + # Inference tab def inference_tab(): default_weight = names[0] if names else None