Update app.py

app.py

@@ -34,7 +34,12 @@ def voice_change(audio_in, audio_ref):
     matching_set = knn_vc.get_matching_set(["./audio_ref.wav"])
     out_wav = knn_vc.match(query_seq, matching_set, topk=4)
     torchaudio.save('output.wav', out_wav[None], 16000)
-    return 'output.wav'
+    noisy = enhance_model.load_audio(
+        'output.wav'
+    ).unsqueeze(0)
+    enhanced = enhance_model.enhance_batch(noisy, lengths=torch.tensor([1.]))
+    torchaudio.save('enhanced.wav', enhanced.cpu(), 16000)
+    return 'enhanced.wav'
 
 
 def tts(text, model, voice, api_key):
@@ -61,14 +66,7 @@ def tts(text, model, voice, api_key):
 
     # Get the file path of the temp file
     temp_file_path = temp_file.name
-    
-    noisy = enhance_model.load_audio(
-        temp_file_path
-    ).unsqueeze(0)
-    enhanced = enhance_model.enhance_batch(noisy, lengths=torch.tensor([1.]))
-    torchaudio.save('enhanced.wav', enhanced.cpu(), 16000)
-    return 'enhanced.wav'
-    
+        
     return temp_file_path