diff --git a/backend/Dockerfile b/backend/Dockerfile
index c577906..c7d79c0 100644
--- a/backend/Dockerfile
+++ b/backend/Dockerfile
@@ -1,7 +1,8 @@
-FROM bitnami/pytorch:2.2.0-debian-11-r15
+FROM python:3.10
 RUN sudo apt-get update \
  && sudo apt-get install -y libgl1-mesa-glx libgtk2.0-0 libsm6 libxext6 
 WORKDIR /home/app
+RUN pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 COPY . .
diff --git a/backend/requirements.txt b/backend/requirements.txt
index 0ebef23..0b0e400 100644
--- a/backend/requirements.txt
+++ b/backend/requirements.txt
@@ -3,6 +3,4 @@
 Flask==3.0.2
 numpy==1.26.4
 opencv-python==4.9.0.80
-torch==2.2.0
-torchvision==0.17.0
 pytest==8.0.0
\ No newline at end of file