diff --git a/Dockerfile.cpu b/Dockerfile.cpu
index e9f2d3b..04c3b8a 100644
--- a/Dockerfile.cpu
+++ b/Dockerfile.cpu
@@ -11,8 +11,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     && rm -rf /var/lib/apt/lists/*
 
 # Install PyTorch CPU version and ONNX runtime
-RUN pip3 install --no-cache-dir torch==2.5.1 --extra-index-url https://download.pytorch.org/whl/cpu && \
-    pip3 install --no-cache-dir onnxruntime==1.20.1
+RUN pip3 install --no-cache-dir torch==2.5.1 --extra-index-url https://download.pytorch.org/whl/cpu 
 
 # Install all other dependencies from requirements.txt
 COPY requirements.txt .
diff --git a/requirements.txt b/requirements.txt
index 154e77e..b71e9c2 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -11,6 +11,7 @@ sqlalchemy==2.0.27
 transformers==4.47.1
 numpy==2.2.1
 scipy==1.14.1
+onnxruntime==1.20.1
 
 # Audio processing
 soundfile==0.13.0