22 lines
307 B
Docker
22 lines
307 B
Docker
FROM python:3.11-slim
|
|
|
|
ARG TASK_TYPE
|
|
ARG GPU_TYPE
|
|
ARG FRAMEWORK
|
|
|
|
ENV TASK_TYPE=${TASK_TYPE}
|
|
ENV GPU_TYPE=${GPU_TYPE}
|
|
ENV FRAMEWORK=${FRAMEWORK}
|
|
ENV PYTHONUNBUFFERED=1
|
|
|
|
WORKDIR /app
|
|
|
|
COPY requirements.txt .
|
|
RUN pip install --no-cache-dir -r requirements.txt
|
|
|
|
COPY . .
|
|
|
|
EXPOSE 8080
|
|
|
|
CMD ["python", "main.py"]
|