Update app.py

app.py

@@ -22,6 +22,9 @@ config = BarkConfig()
 model = Bark.init_from_config(config)
 model.load_checkpoint(config, checkpoint_dir="checkpoints/bark", eval=True)
 
+from TTS.api import TTS
+tts = TTS("tts_models/multilingual/multi-dataset/bark", gpu=True)
+
 def infer(prompt, input_wav_file):
 
     print("SAVING THE AUDIO FILE TO WHERE IT BELONGS")
@@ -44,6 +47,7 @@ def infer(prompt, input_wav_file):
     # Move the WAV file to the new directory
     shutil.move(source_path, os.path.join(destination_path, f"{file_name}.wav"))
 
+    """
     text = prompt
 
     print("SYNTHETIZING...")
@@ -62,6 +66,8 @@ def infer(prompt, input_wav_file):
     
     print(output_dict)
 
+    
+
     sample_rate = 24000  # Replace with the actual sample rate
     print("WRITING WAVE FILE")
     wavfile.write(
@@ -69,6 +75,12 @@ def infer(prompt, input_wav_file):
         sample_rate, 
         output_dict['wav']
     )
+    """
+    
+    tts.tts_to_file(text=prompt,
+                file_path="output.wav",
+                voice_dir="bark_voices/",
+                speaker=f"{file_name}")
 
     # List all the files and subdirectories in the given directory
     contents = os.listdir(f"bark_voices/{file_name}")