Update inference.py

inference.py

@@ -24,12 +24,13 @@ gdown.download(f"{url}{file_id}", output, quiet=False)
 fake_voice=TTS(model_path=f"{cur_path}/best_model.pth",
                config_path=f"{cur_path}/config.json")
 
-def text_to_speech(text="Hola soy gustavo petro y esto es una prueba",output_name="test_audio.wav",model=fake_voice):
+def text_to_speech(text="Hola soy gustavo petro y esto es una prueba",use_nr=True,output_name="test_audio.wav",model=fake_voice):
 
     model.tts_to_file(text,file_path=output_name)
     rate, data = wavfile.read(output_name)
-    reduced_noise = nr.reduce_noise(y=data, sr=rate)
-    wavfile.write(output_name, rate, reduced_noise)
+    if use_nr:
+        reduced_noise = nr.reduce_noise(y=data, sr=rate)
+        wavfile.write(output_name, rate, reduced_noise)
     return output_name