自动加载最新模型

app.py

@@ -58,6 +58,7 @@ device = (
     )
 
 BandList = {
+        #"PoppinParty":["香澄","有咲","たえ","りみ","沙綾"],
         "Afterglow":["蘭","モカ","ひまり","巴","つぐみ"],
         "HelloHappyWorld":["こころ","美咲","薫","花音","はぐみ"],
         "PastelPalettes":["彩","日菜","千聖","イヴ","麻弥"],
@@ -213,17 +214,17 @@ if __name__ == "__main__":
     REPO_ID = "audeering/wav2vec2-large-robust-12-ft-emotion-msp-dim"
     emotional_processor = Wav2Vec2Processor.from_pretrained(emotional_model_name)
     emotional_model = EmotionModel.from_pretrained(emotional_model_name).to(device)
-    hps = utils.get_hparams_from_file('Data/Bushiroad/configs/config.json')
-    net_g = get_net_g(
-        model_path='Data/Bushiroad/models/G_29000.pth', version="2.1", device=device, hps=hps
-    )
-    speaker_ids = hps.data.spk2id
-    speakers = list(speaker_ids.keys())
     languages = [ "Auto", "ZH", "JP"]
     modelPaths = []
     for dirpath, dirnames, filenames in os.walk("Data/Bushiroad/models/"):
         for filename in filenames:
             modelPaths.append(os.path.join(dirpath, filename))
+    hps = utils.get_hparams_from_file('Data/Bushiroad/configs/config.json')
+    net_g = get_net_g(
+        model_path=modelPaths[-1], version="2.1", device=device, hps=hps
+    )
+    speaker_ids = hps.data.spk2id
+    speakers = list(speaker_ids.keys())
     with gr.Blocks() as app:
         for band in BandList:
             with gr.TabItem(band):
@@ -298,4 +299,4 @@ if __name__ == "__main__":
                     )
 
     print("推理页面已开启!")
-    app.launch()
+    app.launch(share=True)