Update coqui.py

coqui.py

@@ -88,7 +88,7 @@ model.load_checkpoint(
     checkpoint_dir=model_path,
     vocab_path=vocab_path,
     eval=True,
-    use_deepspeed=True,
+    use_deepspeed=False,
 )
 model.cuda()
 
@@ -133,6 +133,15 @@ def predict(
     no_lang_auto_detect,
     agree,
 ):
+    print("####################################### Predict Called ##############################")
+    print("promp:",prompt)
+    print("language:",language)
+    print("audio_file_pth:",audio_file_pth)
+    print("mic_file_path:",mic_file_path)
+    print("use_mic:",use_mic)
+    print("voice_cleanup:",voice_cleanup)
+    print("no_lang_auto_detect:",no_lang_auto_detect)
+    print("agree:",agree)
     if agree == True:
         if language not in supported_languages:
             print(
@@ -321,7 +330,7 @@ def predict(
                 # Convert chunk to numpy array and return it
                 chunk_np = chunk.cpu().numpy()
                 print('chunk',i)
-                yield (24000, chunk_np)
+                yield numpy_to_mp3(chunk_np,24000)
                 wav_chunks.append(chunk)
                 
                 print(f"Received chunk {i} of audio length {chunk.shape[-1]}")
@@ -415,4 +424,4 @@ def predict(
         print("Please accept the Terms & Condition!")
         return (
             None,
-        )
+        )