Den4ikAI commited on
Commit
479251c
·
verified ·
1 Parent(s): dc0e249

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +33 -0
app.py ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from runorm import RUNorm
3
+
4
+ models = {
5
+ "RUNorm-small": "small",
6
+ "RUNorm-medium": "medium",
7
+ "RUNorm-big": "big"
8
+ }
9
+
10
+ normalizers = {}
11
+
12
+ for model_name, model_size in models.items():
13
+ normalizer = RUNorm()
14
+ normalizer.load(model_size=model_size, device="cpu")
15
+ normalizers[model_name] = normalizer
16
+
17
+ def normalize_text(input_text, model_name):
18
+ normalizer = normalizers[model_name]
19
+ normalized_text = normalizer.norm(input_text)
20
+ return normalized_text
21
+
22
+ iface = gr.Interface(
23
+ fn=normalize_text,
24
+ inputs=[
25
+ gr.components.Textbox(lines=5, placeholder="Введите текст для нормализации"),
26
+ gr.components.Radio(list(models.keys()), label="Модель", value="RUNorm-medium")
27
+ ],
28
+ outputs="text",
29
+ title="RUNorm",
30
+ description="Демо RUNorm - модели для нормализации русского текста перед синтезом речи.",
31
+ )
32
+
33
+ iface.launch()