speaker-detector 0.1.4__py3-none-any.whl → 0.1.5__py3-none-any.whl

speaker_detector/export_embeddings.py

@@ -39,3 +39,24 @@ def export_embeddings_to_json(pt_path, json_path):
         json.dump(converted, f, indent=2)
 
     print(f"✅ Exported {len(converted)} speaker embeddings to {json_path}")
+
+def load_embeddings_from_json(json_path, embeddings_dir):
+    """Load speaker embeddings directly from uploaded JSON format."""
+    with open(json_path, "r") as f:
+        speakers = json.load(f)
+
+    if not isinstance(speakers, list):
+        raise ValueError("Expected a list of speaker records in speakers.json")
+
+    for entry in speakers:
+        speaker_id = entry.get("label")
+        vector = entry.get("vector")
+
+        if speaker_id and isinstance(vector, list):
+            emb_path = embeddings_dir / f"{speaker_id}.pt"
+            torch.save(torch.tensor(vector), emb_path)
+            print(f"✅ Loaded embedding for {speaker_id} → {emb_path}")
+        else:
+            print(f"⚠️ Skipping invalid speaker record: {entry}")
+
+

{speaker_detector-0.1.4.dist-info → speaker_detector-0.1.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: speaker-detector
-Version: 0.1.4
+Version: 0.1.5
 Summary: A CLI tool for speaker enrollment and identification using SpeechBrain.
 Author-email: Lara Whybrow <lara.whybrow@gmail.com>
 License: MIT

{speaker_detector-0.1.4.dist-info → speaker_detector-0.1.5.dist-info}/RECORD

@@ -5,11 +5,11 @@ speaker_detector/analyze.py,sha256=sA8qyzczdHUbJw2_1JIbXn1WpiKC5dHLPRtPPoppJzY,1
 speaker_detector/cli.py,sha256=TKci4o4Fru-3NqUkPDRQRvtis2niNEAh9sQWwE5t6Us,3521
 speaker_detector/combine.py,sha256=yCiqG6VMojz0CxSTPqjx0RrUban8oFIcKlA1zFMzaU4,761
 speaker_detector/core.py,sha256=lQNOcmZs2IJOqrNKlk1BeVQX6tzc7BSpeP5Gordff-E,3586
-speaker_detector/export_embeddings.py,sha256=OxNXadzEiMEJgpmCG6HHFncUX7DumFvTOys1R6UMUnw,1151
+speaker_detector/export_embeddings.py,sha256=PJtj_tdas595fnjRybnyCVMG7T84-xgDYPp7eKvpKnw,1922
 speaker_detector/export_model.py,sha256=qVVT2wSCnsPA8pSAEEyIMkY7Kc8uAgepc03MxBMT3xU,1146
 speaker_detector/generate_summary.py,sha256=oTWEf2bxTCRIUl8L17-J64FyhRbCPnDjihFluEnBWc8,3726
-speaker_detector-0.1.4.dist-info/METADATA,sha256=c0l2KJOvthuvSJ_g7QLb23sYDHqXhm4wi8dJYChEB8M,4564
-speaker_detector-0.1.4.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
-speaker_detector-0.1.4.dist-info/entry_points.txt,sha256=2B30ee2cTyeeA49x_TBURl53bDRiLWGK3NWhb9rlK3s,63
-speaker_detector-0.1.4.dist-info/top_level.txt,sha256=PJ5rfvd3GAbzMbc7-Fwhtufjf6HxzzTiiHociOy7RiM,17
-speaker_detector-0.1.4.dist-info/RECORD,,
+speaker_detector-0.1.5.dist-info/METADATA,sha256=NJazUcFKkbwoUCSFgfGDT7rjs8cbHqpCG26Y3RQ5zUs,4564
+speaker_detector-0.1.5.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+speaker_detector-0.1.5.dist-info/entry_points.txt,sha256=2B30ee2cTyeeA49x_TBURl53bDRiLWGK3NWhb9rlK3s,63
+speaker_detector-0.1.5.dist-info/top_level.txt,sha256=PJ5rfvd3GAbzMbc7-Fwhtufjf6HxzzTiiHociOy7RiM,17
+speaker_detector-0.1.5.dist-info/RECORD,,