From c68701da4b401f475a0bd730de277c839395cb97 Mon Sep 17 00:00:00 2001
From: "Timothy J. Baek" <timothyjrbeck@gmail.com>
Date: Thu, 30 May 2024 20:01:33 -0700
Subject: [PATCH] fix: tf

---
 Dockerfile       | 3 ---
 requirements.txt | 1 -
 2 files changed, 4 deletions(-)

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