def __init__()

in src/speechbox/restore.py [0:0]

• 5 lines of code
• 1 McCabe index (conditional complexity)

    def __init__(self, model: WhisperForConditionalGeneration, processor: WhisperProcessor):
        self.processor = processor
        self.tokenizer = processor.tokenizer
        self.model = model
        self.punctuation = self.get_punctuation_tokens()