Update app.py

app.py

@@ -52,16 +52,9 @@ def voice_change(audio_in, audio_ref):
 
     query_seq = knn_vc.get_features(audio_in_path)
     matching_set = knn_vc.get_matching_set([audio_ref_path])
-    print("query_seq shape:", query_seq.shape)
-    print("matching_set shape:", matching_set.shape)
     out_wav = knn_vc.match(query_seq, matching_set, topk=4)
-
-    # 确保 out_wav 是二维张量
-    if len(out_wav.shape) == 1:
-        out_wav = out_wav.unsqueeze(0)
-
     output_path = 'output.wav'
-    torchaudio.save(output_path, out_wav, 16000)
+    torchaudio.save(output_path, out_wav[None], 16000)
     return output_path
 
 # 文字转语音（OpenAI）