Update TextGen/router.py

TextGen/router.py

@@ -8,7 +8,7 @@ from fastapi.middleware.cors import CORSMiddleware
 from langchain.chains import LLMChain
 from langchain.prompts import PromptTemplate
 from TextGen.suno import custom_generate_audio, get_audio_information
-#from TextGen.coqui import predict
+from TextGen.coqui import predict
 from langchain_google_genai import (
     ChatGoogleGenerativeAI,
     HarmBlockThreshold,
@@ -165,7 +165,7 @@ async def generate_voice(message: VoiceMessage):
 
         # Generator function to yield audio chunks
         async def audio_stream():
-            result = tts_client.predict(
+            result = predict(
                 prompt=message.input,
                 language=message.language,
                 audio_file_pth=audio_file_pth,