diff --git a/Dockerfile b/Dockerfile
index 82f7c3d..95da76a 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -25,12 +25,9 @@ WORKDIR /app
 # Install Python dependencies
 RUN pip3 install uv && \
     if [ "$USE_CUDA" = "true" ]; then \
-    pip3 install --no-binary=h5py h5py --no-cache-dir && \
     pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/$USE_CUDA_DOCKER_VER --no-cache-dir && \
-    pip3 install tensorflow[and-cuda] --no-cache-dir && \
     uv pip install --system -r requirements.txt --no-cache-dir; \
     else \
-    pip3 install --no-binary=h5py h5py --no-cache-dir && \
     pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu --no-cache-dir && \
     uv pip install --system -r requirements.txt --no-cache-dir; \
     fi
diff --git a/requirements.txt b/requirements.txt
index f6e5e37..dc45638 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -34,7 +34,6 @@ langchain-community
 langfuse
 
 # ML libraries
-tensorflow
 torch
 xgboost