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