diff --git a/bird_cnn/Dockerfile b/bird_cnn/Dockerfile
index 690a1d2..e8bf591 100644
--- a/bird_cnn/Dockerfile
+++ b/bird_cnn/Dockerfile
@@ -14,6 +14,8 @@ COPY . .
 # Install Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
+RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
+
 # Make sure the startup script is executable
 RUN chmod +x start_server.sh
 
diff --git a/bird_cnn/requirements.txt b/bird_cnn/requirements.txt
index 0924cf2..f479b01 100644
--- a/bird_cnn/requirements.txt
+++ b/bird_cnn/requirements.txt
@@ -1,7 +1,5 @@
 fastapi==0.136.1
 networkx==3.6.1
 numpy==2.3.4
-torch==2.1.2
-torchvision==0.16.2
 tqdm==4.67.3
 uvicorn==0.46.0
diff --git a/cnn_website/src/Bird_CNN.jsx b/cnn_website/src/Bird_CNN.jsx
index bd5a62e..d5b9d49 100644
--- a/cnn_website/src/Bird_CNN.jsx
+++ b/cnn_website/src/Bird_CNN.jsx
@@ -4,7 +4,7 @@ import './Bird_CNN.css'
 function Bird_CNN() {
     const apiUrl = process.env.NODE_ENV === "development"
         ? "http://127.0.0.1:8000"
-        : "https://api.myapp.com";
+        : "http://134.60.154.7:8000";
 
     const [file, setFile] = useState(null);
     const [birdClass, setBirdClass] = useState(null);