Update app.py

app.py

@@ -204,7 +204,7 @@ if __name__ == "__main__":
     emotional_model = EmotionModel.from_pretrained(emotional_model_name).to(device)
     hps = utils.get_hparams_from_file('Data/BanGDream/configs/config.json')
     net_g = get_net_g(
-        model_path='Data/BanGDream/models/G_63000.pth', version="2.1", device=device, hps=hps
+        model_path='Data/BanGDream/models/G_132000.pth', version="2.1", device=device, hps=hps
     )
     speaker_ids = hps.data.spk2id
     speakers = list(speaker_ids.keys())
@@ -218,6 +218,10 @@ if __name__ == "__main__":
             with gr.TabItem(band):
                 for name in BandList[band]:
                     with gr.TabItem(name):
+                        classifiedPaths = []
+                        for dirpath, dirnames, filenames in os.walk("Data/BanGDream/classifedSample/"+name):
+                            for filename in filenames:
+                                classifiedPaths.append(os.path.join(dirpath, filename))
                         with gr.Row():
                             with gr.Column():
                                 with gr.Row():
@@ -256,7 +260,7 @@ if __name__ == "__main__":
                                     placeholder="输入纯日语或者中文",
                                     value="为什么要演奏春日影!",
                                 )
-                                reference_audio = gr.Audio(label="情感参考音频（WAV 格式），从(此处)[https://huggingface.co/spaces/Mahiruoshi/MyGO_VIts-bert/tree/main/Data/BanGDream/TempInput]获取", type="filepath")                                
+                                reference_audio = gr.Dropdown(label = "情感参考", choices = classifiedPaths, value = classifiedPaths[0], type = "value")
                                 btn = gr.Button("点击生成", variant="primary")
                                 audio_output = gr.Audio(label="Output Audio")
                                 '''