Create Dockerfile

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"]