changed dockerfile

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
 

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