@huggingface/transformers/src/models/chatterbox/processing_chatterbox.js

State-of-the-art Machine Learning for the web. Run 🤗 Transformers directly in your browser, with no need for a server!

import { AutoFeatureExtractor } from '../auto/feature_extraction_auto.js';
import { AutoTokenizer } from '../auto/tokenization_auto.js';
import { Processor } from '../../processing_utils.js';

/**
 * Represents a ChatterboxProcessor that extracts features from an audio input.
 */
export class ChatterboxProcessor extends Processor {
    static tokenizer_class = AutoTokenizer;
    static feature_extractor_class = AutoFeatureExtractor;

    async _call(text, audio = null) {
        const text_features = this.tokenizer(text);
        const audio_features = audio ? await this.feature_extractor(audio) : {};
        return { ...text_features, ...audio_features };
    }
}