# This Dockerfile serves as the build file for Huggingface Spaces
FROM huggingface/transformers-pytorch-gpu

ARG ML_APP_LISTEN_PORT=7860
ARG ML_MLFLOW_ENDPOINT
ARG ML_HF_ACCESS_TOKEN

ENV APP_LISTEN_PORT=${ML_APP_LISTEN_PORT}
ENV MLFLOW_ENDPOINT=${ML_MLFLOW_ENDPOINT}
ENV HF_ACCESS_TOKEN=${ML_HF_ACCESS_TOKEN}

RUN apt-get update
RUN /usr/bin/python3 -m pip install uvicorn fastapi mlflow huggingface_hub httpx

WORKDIR /app

COPY ./app /app

EXPOSE ${ML_APP_LISTEN_PORT}

ENTRYPOINT [ "/usr/bin/python3", "/app/main.py" ]