# Load model directly
from transformers import AutoProcessor, AutoModelForSpeechSeq2Seq

processor = AutoProcessor.from_pretrained("pedrogarcias/whisper-small-v2-jax")
model = AutoModelForSpeechSeq2Seq.from_pretrained("pedrogarcias/whisper-small-v2-jax")