diff options
Diffstat (limited to 'deep_phonemizer/app.py')
| -rw-r--r-- | deep_phonemizer/app.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/deep_phonemizer/app.py b/deep_phonemizer/app.py index ab5626c..b065781 100644 --- a/deep_phonemizer/app.py +++ b/deep_phonemizer/app.py @@ -2,6 +2,11 @@ from dp.phonemizer import Phonemizer from flask import Flask from flask import request + +import torch +from dp.preprocessing.text import Preprocessor, LanguageTokenizer, SequenceTokenizer +torch.serialization.add_safe_globals([Preprocessor, LanguageTokenizer, SequenceTokenizer]) + phonemizer = Phonemizer.from_checkpoint('assets/model.pt') app = Flask(__name__, instance_relative_config=True) |