src/model/feature_extrator.py

import torch
from transformers import Wav2Vec2Model, Wav2Vec2Processor
from config import MODEL_NAME, DEVICE

processor = Wav2Vec2Processor.from_pretrained(MODEL_NAME)
feature_extractor = Wav2Vec2Model.from_pretrained(MODEL_NAME).to(DEVICE)