Update tts.py

tts.py

@@ -6,6 +6,8 @@ import ffmpeg
 import shutil
 import argparse
 
+os.environ["COQUI_TOS_AGREED"] = "1"
+
 def adjust_speed(input_file, speed_factor):
     output_file = input_file.replace(".wav", "_adjusted.wav")
     ffmpeg.input(input_file).filter('atempo', speed_factor).output(output_file, acodec='pcm_s16le').run()
@@ -87,8 +89,6 @@ def main(speaker_directory, aligned_text_file, output_audio_file):
         shutil.rmtree('./audio/temp')
 
 if __name__ == "__main__":
-    os.environ["COQUI_TOS_AGREED"] = "1"
-    
     parser = argparse.ArgumentParser(description="Generate speech from translated text")
     parser.add_argument("speaker_directory", help="Directory containing speaker voice clips")
     parser.add_argument("aligned_text_file", help="Path to the translated and aligned text file")