device cuda; modify header

app.py

@@ -14,10 +14,11 @@ from text import text_to_sequence
 import gradio as gr
 
 
-pth_path = "G_240000.pth"
+pth_path = os.path.basename(utils.latest_checkpoint_path("./", "G_*.pth"))
+# pth_path = "G_250000.pth"
 hps = utils.get_hparams_from_file("./configs/hoshimi_base.json")
-# device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
-device = torch.device("cpu")
+device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+# device = torch.device("cpu")
 
 def get_text(text, hps):
     text_norm = text_to_sequence(text, hps.data.text_cleaners)
@@ -47,8 +48,6 @@ def list_model():
             continue
         if (f.endswith(".pth")):
             res.append(f)
-            if len(f) >= len(pth_path):
-                pth_path = f
     return res
 
 
@@ -66,9 +65,9 @@ net_g = load_model(pth_path)
 
 def change_model(model):
     global pth_path
-    global net_g_ms
+    global net_g
     pth_path = model
-    net_g_ms = load_model(pth_path)
+    net_g = load_model(pth_path)
     return "载入模型："+pth_path
 
 
@@ -83,7 +82,7 @@ with app:
             tts_input1 = gr.TextArea(
                 label="请输入文本（目前只支持汉字和单个英文字母，也可以使用逗号、句号、感叹号、空格等常用符号来改变语调和停顿）",
                 value="这里是爱喝奶茶，穿得也像奶茶魅力点是普通话二乙的星弥吼西咪，晚上齁。")
-            tts_submit = gr.Button("用文本合成", variant="primary")
+            tts_submit = gr.Button("合成", variant="primary")
             tts_output = gr.Audio(label="Output")
             tts_model = gr.Markdown("")
             tts_submit.click(infer, [tts_input1], [tts_output])

header.html

@@ -1,5 +1,5 @@
 <div
-    style="width: 100%;padding-top:116px;background-image: url('https://huggingface.co/spaces/candlend/vits-hoshimi/resolve/main/header.png');;background-size:cover">
+    style="width: 100%;padding-top:116px;background-image: url('https://huggingface.co/spaces/candlend/vits-hoshimi/resolve/main/header.png');background-size:cover;background-position:right;">
     <div>
         <div style="margin: 0px 20px;display: flex;">
             <div class="bili-avatar" style="padding-top: 20px;">
@@ -9,12 +9,12 @@
                 </a>
             </div>
             <div style="margin:20px;color:white">
-                <div style="align-items: flex-end;display: flex">
-                    <span style="font-size:20px;min-width:85px;color:orange">星弥Hoshimi</span>
+                <div style="align-items: flex-end; display: flex">
+                    <span style="font-size:20px;min-width:85px;color:black">星弥Hoshimi</span>
                     <img style="padding-left: 3px;" title="粉丝数"
                         src="https://img.shields.io/badge/dynamic/json?color=orange&label=%E7%B2%89%E4%B8%9D%E6%95%B0&query=data.follower&url=https%3A%2F%2Fapi.bilibili.com%2Fx%2Frelation%2Fstat%3Fvmid%3D477342747"></img>
                     <img style="padding-left: 5px;"
-                        src="https://img.shields.io/badge/VirtuaReal-%E4%BA%94%E6%9C%9F%E7%94%9F-orange"
+                        src="https://img.shields.io/badge/VR-%E4%BA%94%E6%9C%9F%E7%94%9F-orange"
                         title="五期生"></img>
                 </div>
             </div>