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