blob: b065781ec09be92f64950ebe36c1042b59a533bd (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
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)
@app.route('/')
def handle():
"""
Simple route that handles the phonem to grapheme translation.
"""
grapheme = request.args.get('grapheme')
if grapheme is None:
return "You are missing the 'grapheme' parameter", 400
lang = request.args.get('language')
if lang is None:
lang = 'fr'
return phonemizer(grapheme, lang = lang), 200
|
