From 0cf3736b67819b0d3a7c15f13613da01fdc23f45 Mon Sep 17 00:00:00 2001 From: Matthieu Pignolet Date: Wed, 30 Apr 2025 19:05:09 +0400 Subject: 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 --- deep_phonemizer/app.py | 5 +++++ 1 file changed, 5 insertions(+) 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) -- cgit v1.2.3