README.md

Kubeflow Wav2Vec2 Docker Image

Pre-built Docker image with Wav2Vec2 model for audio embeddings in Kubeflow pipelines.

Image Details

• Base: Python 3.10-slim
• Pre-downloaded model: facebook/wav2vec2-base-960h
• Docker Hub: pranav14916/wav2vec2-embeddings:latest

Usage

@dsl.component(
    base_image="pranav14916/wav2vec2-embeddings:latest",
    packages_to_install=[]
)
def generate_audio_embeddings(...):
    # Model already downloaded!