更新了参数调整功能

app.py

@@ -34,7 +34,7 @@ net_g = SynthesizerTrn(
     **hps.model)
 _ = net_g.eval()
 
-_ = utils.load_checkpoint("pretrained_models/G_1153000.pth", net_g, None)
+_ = utils.load_checkpoint("pretrained_models/uma_1153000.pth", net_g, None)
 
 title = "Umamusume voice synthesizer \n 赛马娘语音合成器"
 description = """
@@ -49,7 +49,7 @@ article = """
 """
 
 
-def infer(text, character, language):
+def infer(text, character, language, duration, noise_scale, noise_scale_w):
     if language == '日本語':
         pass
     elif language == '简体中文':
@@ -62,7 +62,7 @@ def infer(text, character, language):
         x_tst = stn_tst.unsqueeze(0)
         x_tst_lengths = torch.LongTensor([stn_tst.size(0)])
         sid = torch.LongTensor([char_id])
-        audio = net_g.infer(x_tst, x_tst_lengths, sid=sid, noise_scale=0.667, noise_scale_w=0.8, length_scale=1)[0][0,0].data.cpu().float().numpy()
+        audio = net_g.infer(x_tst, x_tst_lengths, sid=sid, noise_scale=noise_scale, noise_scale_w=noise_scale_w, length_scale=duration)[0][0,0].data.cpu().float().numpy()
     return (text,(22050, audio))
 
 # We instantiate the Textbox class
@@ -90,16 +90,16 @@ char_dropdown = gr.Dropdown(['0:特别周','1:无声铃鹿','2:东海帝王','3:
                             '76:也文摄辉','77:吉兆','78:谷野美酒','79:第一红宝石',
                             '80:真弓快车','81:骏川手纲','82:凯斯奇迹','83:小林历奇',
                             '84:北港火山','85:奇锐骏','86:秋川理事长'])
-language_dropdown = gr.Dropdown(['日本語','简体中文','English'])
-examples = [['お疲れ様です，トレーナーさん。', '1:无声铃鹿', '日本語'],
-            ['張り切っていこう！', '67:北部玄驹', '日本語'],
-            ['何でこんなに慣れでんのよ，私のほが先に好きだっだのに。', '10:草上飞','日本語'],
-            ['授業中に出しだら，学校生活終わるですわ。', '12:目白麦昆','日本語'],
-            ['お帰りなさい，お兄様！', '29:米浴','日本語'],
-            ['私の処女をもらっでください！', '29:米浴','日本語']]
+language_dropdown = gr.Dropdown(['日本語','简体中文','English', 1, 0.667, 0.8])
+examples = [['お疲れ様です，トレーナーさん。', '1:无声铃鹿', '日本語', 1, 0.667, 0.8],
+            ['張り切っていこう！', '67:北部玄驹', '日本語', 1, 0.667, 0.8],
+            ['何でこんなに慣れでんのよ，私のほが先に好きだっだのに。', '10:草上飞','日本語', 1, 0.667, 0.8],
+            ['授業中に出しだら，学校生活終わるですわ。', '12:目白麦昆','日本語', 1, 0.667, 0.8],
+            ['お帰りなさい，お兄様！', '29:米浴','日本語', 1, 0.667, 0.8],
+            ['私の処女をもらっでください！', '29:米浴','日本語', 1, 0.667, 0.8]]
 
 duration_slider = gr.Slider(minimum=0.1, maximum=5, value=1, step=0.1, label='时长 Duration')
 noise_scale_slider = gr.Slider(minimum=0.1, maximum=5, value=0.667, step=0.001, label='噪声比例 noise_scale')
 noise_scale_w_slider = gr.Slider(minimum=0.1, maximum=5, value=0.8, step=0.1, label='噪声偏差 noise_scale_w')
 
-gr.Interface(fn=infer, inputs=[textbox, char_dropdown, language_dropdown], outputs=["text","audio"],title=title, description=description, article=article, examples=examples).launch()
+gr.Interface(fn=infer, inputs=[textbox, char_dropdown, language_dropdown, duration_slider, noise_scale_slider, noise_scale_w_slider,], outputs=["text","audio"],title=title, description=description, article=article, examples=examples).launch()