Update main.py

main.py

@@ -6,22 +6,21 @@ from fastapi import FastAPI
 from fastapi.staticfiles import StaticFiles
 from pydantic import BaseModel
 from transformers import pipeline
-from kokoro import KPipeline
+from kokoro_onnx import Kokoro
 
 app = FastAPI()
 
-# Setup static folder
 if not os.path.exists("static"):
     os.makedirs("static", exist_ok=True)
 app.mount("/static", StaticFiles(directory="static"), name="static")
 
-# Load Models
+# Load SmolLM2 for Text
 print("Loading SmolLM2...")
 chat_pipe = pipeline("text-generation", model="HuggingFaceTB/SmolLM2-360M-Instruct", device="cpu")
 
-print("Loading Kokoro...")
-# 'a' stands for American English
-tts_pipe = KPipeline(lang_code='a')
+# Load Kokoro ONNX (Bypasses misaki/espeak errors)
+print("Loading Kokoro ONNX...")
+kokoro = Kokoro("kokoro-v1.0.onnx", "voices.bin")
 
 class ChatRequest(BaseModel):
     message: str
@@ -29,21 +28,19 @@ class ChatRequest(BaseModel):
 
 @app.post("/chat")
 async def chat_and_tts(request: ChatRequest):
-    # 1. Text Generation
+    # 1. Generate Text
     prompt = f"You are Bella, a smart and witty female AI assistant. Keep it very short.\nUser: {request.message}\nBella:"    
     outputs = chat_pipe(prompt, max_new_tokens=100, temperature=0.3, do_sample=True)
     full_text = outputs[0]['generated_text'].split("Bella:")[-1].split("User:")[0].strip()
     
-    # 2. Audio Generation (Starts only after full_text is complete)
-    generator = tts_pipe(full_text, voice=request.voice, speed=1.0)
-    audio_chunks = [audio for _, _, audio in generator]
-    combined_audio = np.concatenate(audio_chunks)
+    # 2. Generate Audio via ONNX
+    samples, sample_rate = kokoro.create(full_text, voice=request.voice, speed=1.0, lang="en-us")
     
     file_path = "static/output.wav"
-    sf.write(file_path, combined_audio, 24000)
+    sf.write(file_path, samples, sample_rate)
     
-    # 3. Final Response
-    # Change 'shahid202' to your actual space path
+    # 3. Response
+    # Change 'shahid202' to your actual space name if different
     space_url = "https://shahid202-all.hf.space" 
     
     return {