Update app.py

app.py

@@ -24,7 +24,9 @@ app = FastAPI(
 # ------------------------------------------------------------------------------
 # Create one pipeline instance for the entire app.
 model = KModel() # Initialize KModel
-pipeline = StreamKPipeline(lang_code="a", model=model) # Initialize StreamKPipeline, passing the model
+stream_pipeline = StreamKPipeline(lang_code="a", model=model, chunk_size=2) # Initialize StreamKPipeline, passing the model
+pipeline = KPipeline(lang_code="a", model=model) # Initialize KPipeline, passing the model
+
 
 
 # ------------------------------------------------------------------------------
@@ -118,7 +120,7 @@ def tts_streaming(text: str, voice: str = "af_heart", speed: float = 1.0, format
             yield header
 
         # Stream audio chunks from the pipeline.
-        for audio_chunk in pipeline(text=text, voice=voice, speed=speed):
+        for audio_chunk in stream_pipeline(text=text, voice=voice, speed=speed):
             if audio_chunk is not None and audio_chunk.numel() > 0:
                 if format.lower() == "wav":
                     yield audio_chunk_to_pcm_bytes(audio_chunk)
@@ -127,11 +129,10 @@ def tts_streaming(text: str, voice: str = "af_heart", speed: float = 1.0, format
                 else:
                     raise ValueError(f"Unsupported audio format: {format}")
 
-    media_type = "audio/wav" if format.lower() == "wav" else "audio/opus"
 
     return StreamingResponse(
         audio_chunk_generator(),
-        media_type=media_type,
+        media_type="audio/wav",
         headers={"Cache-Control": "no-cache"},
     )