From 779f09af7064999d9b9916364741dec29b67d0f0 Mon Sep 17 00:00:00 2001 From: Filipi Fuchter Date: Fri, 2 May 2025 14:22:38 -0300 Subject: [PATCH] Fixing lint. --- src/pipecat/audio/turn/smart_turn/local_smart_turn.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/pipecat/audio/turn/smart_turn/local_smart_turn.py b/src/pipecat/audio/turn/smart_turn/local_smart_turn.py index f8a8aa91a..a3ee7ebf9 100644 --- a/src/pipecat/audio/turn/smart_turn/local_smart_turn.py +++ b/src/pipecat/audio/turn/smart_turn/local_smart_turn.py @@ -14,7 +14,7 @@ from pipecat.audio.turn.smart_turn.base_smart_turn import BaseSmartTurn try: import torch - from transformers import Wav2Vec2BertForSequenceClassification, AutoFeatureExtractor + from transformers import AutoFeatureExtractor, Wav2Vec2BertForSequenceClassification except ModuleNotFoundError as e: logger.error(f"Exception: {e}") logger.error( @@ -33,7 +33,9 @@ class LocalSmartTurnAnalyzer(BaseSmartTurn): logger.debug("Loading Local Smart Turn model...") # 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 self._turn_processor = AutoFeatureExtractor.from_pretrained(smart_turn_model_path) # Set device to GPU if available, else CPU