diff options
| author | Matthieu Pignolet <m@mpgn.dev> | 2025-04-30 19:05:09 +0400 |
|---|---|---|
| committer | Matthieu Pignolet <m@mpgn.dev> | 2025-04-30 19:05:09 +0400 |
| commit | 0cf3736b67819b0d3a7c15f13613da01fdc23f45 (patch) | |
| tree | 9adb5e299f1ba2ff63254bdd27f6440107a1f278 | |
| parent | cb70e365b15921e1628b9c80d1a5b364b905cfe8 (diff) | |
chore: add preprocessing classes to torch `safe_globals` list
this is requires in newer torch versions for tokenization of inputs/outputs to happen.
Signed-off-by: Matthieu Pignolet <m@mpgn.dev>
| -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) |