diff --git a/Dockerfile b/Dockerfile
index 4a1ebaf..c6784a9 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,20 +1,10 @@
-FROM tensorflow/tensorflow:latest-gpu
+FROM nvidia/cuda:12.2.2-runtime-rockylinux8
 
-WORKDIR /root
+RUN dnf install python39 python39-pip -y
 
-RUN apt-get update
-RUN apt-get install -y make build-essential libssl-dev zlib1g-dev \
-libbz2-dev libreadline-dev libsqlite3-dev wget curl llvm libncurses5-dev \
-libncursesw5-dev xz-utils tk-dev libffi-dev liblzma-dev git -y
-RUN git clone --depth=1 https://github.com/pyenv/pyenv.git .pyenv
-ENV PYTHON_VERSION=3.9
-ENV PYENV_ROOT="$HOME/.pyenv"
-ENV PATH="${PYENV_ROOT}/shims:${PYENV_ROOT}/bin:${PATH}"
-RUN eval "$(pyenv init -)"
-RUN /root/.pyenv/bin/pyenv install ${PYTHON_VERSION}
-RUN /root/.pyenv/bin/pyenv global ${PYTHON_VERSION}
-RUN pip install --upgrade pip
-RUN pip install tflite-model-maker
-RUN pip install numpy==1.23.1 pycocotools roboflow
+WORKDIR /root/app
+COPY ./requirements.txt ./requirements.txt
 
-COPY download.py .
\ No newline at end of file
+RUN python3.9 -m pip install -r requirements.txt
+
+COPY . .
\ No newline at end of file
diff --git a/pyproject.toml b/pyproject.toml
deleted file mode 100644
index 94b04fe..0000000
--- a/pyproject.toml
+++ /dev/null
@@ -1,18 +0,0 @@
-[tool.poetry]
-name = "2023-bunnybots-ml"
-version = "0.1.0"
-description = ""
-authors = ["Your Name <you@example.com>"]
-readme = "README.md"
-
-[tool.poetry.dependencies]
-python = "~3.9.18"
-tflite-model-maker = "^0.4.2"
-numpy = "1.23.1"
-pycocotools = "^2.0.7"
-roboflow = "^1.1.9"
-
-
-[build-system]
-requires = ["poetry-core"]
-build-backend = "poetry.core.masonry.api"
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..14d9c98
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,4 @@
+tflite-model-maker == 0.4.2
+numpy == 1.23.1
+pycocotools == 2.0.7
+roboflow == 1.1.9