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