Spaces:
Runtime error
Runtime error
File size: 402 Bytes
1ee91f8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
# 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])
|