Spaces:
Runtime error
Runtime error
| # text_encoder.py | |
| from src.sbv2.text.symbols import symbols | |
| _symbol_to_id = {s: i for i, s in enumerate(symbols)} | |
| _id_to_symbol = {i: s for i, s in enumerate(symbols)} | |
| def text_to_sequence(text): | |
| sequence = [_symbol_to_id[s] for s in text if s in _symbol_to_id] | |
| return sequence | |
| def sequence_to_text(sequence): | |
| return ''.join([_id_to_symbol[i] for i in sequence if i in _id_to_symbol]) | |