diff --git a/Dockerfile b/Dockerfile
new file mode 100644
index 0000000..1624570
--- /dev/null
+++ b/Dockerfile
@@ -0,0 +1,8 @@
+FROM python:3.11-slim
+WORKDIR /app
+COPY . .
+# Argument for specifying the GPU type, defaulting to "other"
+ARG gpu_type=other
+RUN pip install --no-cache-dir -r requirements_${gpu_type}.txt
+EXPOSE 7851
+CMD ["python", "script.py"]