Update audio_processing/T2A.py

audio_processing/T2A.py

@@ -1,29 +1,26 @@
 import torch
 import soundfile as sf
-from .config import model_mms_tts_eng, tokenizer_mms_tts_eng
+from .config import pipe_tts
+from io import BytesIO
 
 SAMPLING_RATE = 16000
 
 class T2A:
     def __init__(self, input_text: str):
-        self.inputs = tokenizer_mms_tts_eng(input_text, return_tensors="pt")
+        self.output_model = pipe_tts(input_text)
 
     def get_audio(self):
         if self.inputs is not None:   
-            with torch.no_grad():
-                output_model = model_mms_tts_eng(**self.inputs)
+            
+            synth = self.output_model["audio"][0]
 
-            print(f"output_model : {output_model}")
-                    
-            audio = output_model["audio"][0]
-
-            print(f"audio : {audio}")
+            print(f"synth : {synth}")
 
             with BytesIO() as buffer:
-                sf.write(buffer, audio, SAMPLING_RATE, format='wav')
+                sf.write(buffer, synth, SAMPLING_RATE, format='wav')
                 output = buffer.getvalue() # bytes
 
-            print(f"output : {output}")
+            print(f"output : {output}, type : {type(output)}")
 
             return output
         else: