Update app.py

app.py

@@ -149,6 +149,7 @@ def predict(
     chunk_seconds: float = 0.5,
     absolute_thresh: bool = False,
 ):
+    model = models[speaker]
     audio, _ = librosa.load(audio, sr=model.target_sample, duration=duration_limit)
     audio = model.infer_silence(
         audio.astype(np.float32),
@@ -165,6 +166,7 @@ def predict(
     )
     return model.target_sample, audio
 
+
 def predict_song_from_yt(
     ytid_or_url,
     start,
@@ -180,6 +182,7 @@ def predict_song_from_yt(
     chunk_seconds: float = 0.5,
     absolute_thresh: bool = False,
 ):
+    model = models[speaker]
     end = min(start + duration_limit, end)
     original_track_filepath = download_youtube_clip(
         ytid_or_url,
@@ -208,6 +211,8 @@ def predict_song_from_yt(
     full_song = inst_wav + np.expand_dims(cloned_vox, 1)
     return (model.target_sample, full_song), (model.target_sample, cloned_vox)
 
+
+
 description = f"""
 <center>💡 - 如何使用此程序：在页面上方选择“从B站视频上传”模块，填写视频网址和视频起止时间后，点击“submit”按键即可！您还可以点击页面最下方的示例快速预览效果</center>
 """.strip()