Spaces:

bdstar
/

VoiceTutor-AI

Sleeping

bdstar commited on Oct 31

Commit

649294c

verified ·

1 Parent(s): 1c04da9

transcribe only English.

Files changed (1) hide show

app.py CHANGED Viewed

@@ -29,7 +29,7 @@ stt_model = WhisperModel(WHISPER_SIZE, device="cuda" if os.environ.get("CUDA_VIS
                          compute_type="float16" if os.environ.get("CUDA_VISIBLE_DEVICES") else "int8")
 def speech_to_text(audio_path: str) -> str:
-    segments, info = stt_model.transcribe(audio_path, beam_size=1, vad_filter=True)
     text = "".join(seg.text for seg in segments).strip()
     return text

                          compute_type="float16" if os.environ.get("CUDA_VISIBLE_DEVICES") else "int8")
 def speech_to_text(audio_path: str) -> str:
+    segments, info = stt_model.transcribe(audio_path, beam_size=1, vad_filter=True, language="en")
     text = "".join(seg.text for seg in segments).strip()
     return text