diff --git a/Dockerfile.py b/Dockerfile.py
index 3c3836f..1a2a516 100644
--- a/Dockerfile.py
+++ b/Dockerfile.py
@@ -1,9 +1,8 @@
-FROM jupyter/base-notebook
+FROM python:3.10-slim
  
-USER root
-RUN apt-get update && apt-get install -y build-essential g++ libgl1-mesa-glx libx11-6
-COPY requirements.txt .
-RUN pip install -r requirements.txt
-COPY . /src 
 WORKDIR /src
+RUN apt-get update && apt-get install -y build-essential g++ libgl1-mesa-glx libx11-6 && rm -rf /var/lib/apt/lists/*
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+COPY . .