Fix Whisper formatting
This commit is contained in:
@@ -9,10 +9,10 @@
|
||||
import asyncio
|
||||
from enum import Enum
|
||||
from typing import AsyncGenerator, Optional
|
||||
from typing_extensions import TYPE_CHECKING, override
|
||||
|
||||
import numpy as np
|
||||
from loguru import logger
|
||||
from typing_extensions import TYPE_CHECKING, override
|
||||
|
||||
from pipecat.frames.frames import ErrorFrame, Frame, TranscriptionFrame
|
||||
from pipecat.services.ai_services import SegmentedSTTService
|
||||
@@ -332,6 +332,7 @@ class WhisperSTTService(SegmentedSTTService):
|
||||
"""
|
||||
try:
|
||||
from faster_whisper import WhisperModel
|
||||
|
||||
logger.debug("Loading Whisper model...")
|
||||
self._model = WhisperModel(
|
||||
self.model_name, device=self._device, compute_type=self._compute_type
|
||||
@@ -456,10 +457,11 @@ class WhisperSTTServiceMLX(WhisperSTTService):
|
||||
|
||||
whisper_lang = self.language_to_service_language(self._settings["language"])
|
||||
chunk = await asyncio.to_thread(
|
||||
mlx_whisper.transcribe, audio_float,
|
||||
mlx_whisper.transcribe,
|
||||
audio_float,
|
||||
path_or_hf_repo=self.model_name,
|
||||
temperature=self._temperature,
|
||||
language=whisper_lang
|
||||
language=whisper_lang,
|
||||
)
|
||||
text: str = ""
|
||||
for segment in chunk.get("segments", []):
|
||||
|
||||
Reference in New Issue
Block a user