Spaces:
Sleeping
Sleeping
husseinelsaadi Claude Opus 4.8 commited on
Commit ·
145f34b
1
Parent(s): ed131b8
Fix LUNA voice + transcription: edge-tts 7.2.8 (fixes 403), Whisper base model + en/VAD for accuracy
Browse files- backend/services/interview_engine.py +11 -2
- requirements.txt +1 -1
backend/services/interview_engine.py
CHANGED
|
@@ -104,7 +104,7 @@ def load_whisper_model():
|
|
| 104 |
# lightweight model to improve startup times. Available options
|
| 105 |
# include: tiny, base, base.en, small, medium, large. See
|
| 106 |
# https://huggingface.co/ggerganov/whisper.cpp for details.
|
| 107 |
-
model_name = os.getenv("WHISPER_MODEL_NAME", "
|
| 108 |
whisper_model = WhisperModel(model_name, device=device, compute_type=compute_type)
|
| 109 |
logging.info(f"Whisper model '{model_name}' loaded on {device} with {compute_type}")
|
| 110 |
except Exception as e:
|
|
@@ -417,7 +417,16 @@ def whisper_stt(audio_path):
|
|
| 417 |
return ""
|
| 418 |
|
| 419 |
model = load_whisper_model()
|
| 420 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 421 |
transcript = " ".join(segment.text for segment in segments)
|
| 422 |
return transcript.strip()
|
| 423 |
except Exception as e:
|
|
|
|
| 104 |
# lightweight model to improve startup times. Available options
|
| 105 |
# include: tiny, base, base.en, small, medium, large. See
|
| 106 |
# https://huggingface.co/ggerganov/whisper.cpp for details.
|
| 107 |
+
model_name = os.getenv("WHISPER_MODEL_NAME", "base")
|
| 108 |
whisper_model = WhisperModel(model_name, device=device, compute_type=compute_type)
|
| 109 |
logging.info(f"Whisper model '{model_name}' loaded on {device} with {compute_type}")
|
| 110 |
except Exception as e:
|
|
|
|
| 417 |
return ""
|
| 418 |
|
| 419 |
model = load_whisper_model()
|
| 420 |
+
# language="en" avoids misdetecting the language; vad_filter drops
|
| 421 |
+
# silence so Whisper doesn't hallucinate phrases on quiet gaps;
|
| 422 |
+
# condition_on_previous_text=False keeps each answer independent.
|
| 423 |
+
segments, _ = model.transcribe(
|
| 424 |
+
wav_path,
|
| 425 |
+
language="en",
|
| 426 |
+
beam_size=5,
|
| 427 |
+
vad_filter=True,
|
| 428 |
+
condition_on_previous_text=False,
|
| 429 |
+
)
|
| 430 |
transcript = " ".join(segment.text for segment in segments)
|
| 431 |
return transcript.strip()
|
| 432 |
except Exception as e:
|
requirements.txt
CHANGED
|
@@ -24,7 +24,7 @@ accelerate==0.29.3
|
|
| 24 |
huggingface_hub==0.20.3
|
| 25 |
faster-whisper==1.1.1
|
| 26 |
av==13.1.0
|
| 27 |
-
edge-tts==
|
| 28 |
gunicorn
|
| 29 |
python-dotenv
|
| 30 |
chromadb>=0.4.0
|
|
|
|
| 24 |
huggingface_hub==0.20.3
|
| 25 |
faster-whisper==1.1.1
|
| 26 |
av==13.1.0
|
| 27 |
+
edge-tts==7.2.8
|
| 28 |
gunicorn
|
| 29 |
python-dotenv
|
| 30 |
chromadb>=0.4.0
|