diff --git a/backend/Dockerfile b/backend/Dockerfile
index f4c0c99..e7b1693 100644
--- a/backend/Dockerfile
+++ b/backend/Dockerfile
@@ -1,5 +1,6 @@
 # Use an official Python runtime with NVIDIA CUDA support
-FROM nvidia/cuda:11.2.2-cudnn8-runtime-ubuntu20.04
+FROM python:3.9-slim
+
 
 WORKDIR /app
 
@@ -7,8 +8,7 @@ WORKDIR /app
 COPY . /app
 
 # Install Python and FastAPI dependencies
-RUN apt-get update && apt-get install -y python3-pip && \
-    pip3 install --no-cache-dir -r requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
 
 EXPOSE 8000