Add back the spaces
This commit is contained in:
@@ -5,8 +5,10 @@ import torch
|
||||
|
||||
import tts_preprocessor
|
||||
|
||||
|
||||
torch._C._jit_set_profiling_mode(False)
|
||||
|
||||
|
||||
params = {
|
||||
'activate': True,
|
||||
'speaker': 'en_49',
|
||||
@@ -34,15 +36,18 @@ table = str.maketrans({
|
||||
'"': """,
|
||||
})
|
||||
|
||||
|
||||
def xmlesc(txt):
|
||||
return txt.translate(table)
|
||||
|
||||
|
||||
def load_model():
|
||||
model, example_text = torch.hub.load(repo_or_dir='snakers4/silero-models', model='silero_tts', language=params['language'], speaker=params['model_id'])
|
||||
model.to(params['device'])
|
||||
return model
|
||||
model = load_model()
|
||||
|
||||
|
||||
def output_modifier(string):
|
||||
"""
|
||||
This function is applied to the model outputs.
|
||||
@@ -70,6 +75,7 @@ def output_modifier(string):
|
||||
|
||||
print(string)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
import sys
|
||||
output_modifier(sys.argv[1])
|
||||
|
||||
Reference in New Issue
Block a user