| 1 | #!/usr/bin/env python3 |
| 2 | """Enroll a named voice into the library from a recording. |
| 3 | |
| 4 | python enroll.py <audio> [name] # default name: You |
| 5 | """ |
| 6 | import os |
| 7 | import sys |
| 8 | |
| 9 | sys.path.insert(0, os.path.dirname(os.path.abspath(__file__))) |
| 10 | from speaker_id import embed_file, upsert_voice |
| 11 | |
| 12 | audio = sys.argv[1] |
| 13 | name = sys.argv[2] if len(sys.argv) > 2 else "You" |
| 14 | upsert_voice(name, embed_file(audio)) |
| 15 | print(f"enrolled {name}") |