wx44wx/pipecat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixing lint.

Browse Source
This commit is contained in:
Filipi Fuchter
2025-05-02 14:22:38 -03:00
parent 19dc0f2bfb
commit 779f09af70
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/pipecat/audio/turn/smart_turn/local_smart_turn.py
View File

@@ -14,7 +14,7 @@ from pipecat.audio.turn.smart_turn.base_smart_turn import BaseSmartTurn
try: try:
import torch import torch
from transformers import Wav2Vec2BertForSequenceClassification, AutoFeatureExtractor from transformers import AutoFeatureExtractor, Wav2Vec2BertForSequenceClassification
except ModuleNotFoundError as e: except ModuleNotFoundError as e:
logger.error(f"Exception: {e}") logger.error(f"Exception: {e}")
logger.error( logger.error(
@@ -33,7 +33,9 @@ class LocalSmartTurnAnalyzer(BaseSmartTurn):
logger.debug("Loading Local Smart Turn model...") logger.debug("Loading Local Smart Turn model...")
# Load the pretrained model for sequence classification # Load the pretrained model for sequence classification
self._turn_model = Wav2Vec2BertForSequenceClassification.from_pretrained(smart_turn_model_path) self._turn_model = Wav2Vec2BertForSequenceClassification.from_pretrained(
smart_turn_model_path
)
# Load the corresponding feature extractor for preprocessing audio # Load the corresponding feature extractor for preprocessing audio
self._turn_processor = AutoFeatureExtractor.from_pretrained(smart_turn_model_path) self._turn_processor = AutoFeatureExtractor.from_pretrained(smart_turn_model_path)
# Set device to GPU if available, else CPU # Set device to GPU if available, else CPU