Update app.py

app.py

@@ -2,7 +2,7 @@
 from fastapi import FastAPI, UploadFile, File, HTTPException, Request
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
-import whisper
+from faster_whisper import WhisperModel
 from pathlib import Path
 import tempfile, subprocess, os, shutil, base64
 
@@ -22,8 +22,7 @@ DEVICE = "cpu"
 COMPUTE = "int8"
 
 try:
-    MODEL_NAME = "base" 
-    model = whisper.load_model(MODEL_NAME, device=DEVICE)
+    model = WhisperModel(MODEL_NAME, device=DEVICE, compute_type=COMPUTE)
 except Exception as e:
     raise RuntimeError(f"Failed to load Whisper model: {e}")
 
@@ -54,8 +53,9 @@ def _cleanup(paths):
 
 def _do_transcribe(wav_path: str):
     try:
-        result = model.transcribe(wav_path)
-        return {"text": result["text"], "language": result.get("language", "unknown"), "duration": 0}
+        segments, info = model.transcribe(wav_path, vad_filter=True)
+        text = "".join(s.text for s in segments)
+        return {"text": text, "language": info.language, "duration": info.duration}
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"ASR error: {e}")
 
@@ -107,4 +107,4 @@ async def transcribe_b64(payload: AudioB64):
 async def echo(request: Request, file: UploadFile | None = File(default=None)):
     ct = request.headers.get("content-type")
     name = getattr(file, "filename", None) if file else None
-    return {"has_file": file is not None, "filename": name, "content_type": ct}
+    return {"has_file": file is not None, "filename": name, "content_type": ct}