Dockerfile

FROM python:3.9

WORKDIR /app

COPY ./requirements.txt /app/requirements.txt
COPY ./packages.txt /app/packages.txt

RUN apt-get update && xargs -r -a /app/packages.txt apt-get install -y && rm -rf /var/lib/apt/lists/*
RUN pip3 install --no-cache-dir -r /app/requirements.txt
RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
RUN wget -O /app/MIN_RESNET101_BMI_Cache_test.pkl "https://github.com/ManavSarkar/Weight-Prediction-using-Machine-Learning/releases/download/dataset/MIN_RESNET101_BMI_Cache_test.pkl"
RUN wget -O /app/gpr_model_withgender.pkl "https://github.com/ManavSarkar/Weight-Prediction-using-Machine-Learning/releases/download/dataset/gpr_model_withgender.pkl"
COPY detectron2 /app/detectron2

RUN pip install -e /app/detectron2
# User
RUN useradd -m -u 1000 user
USER user
ENV HOME /home/user
ENV PATH $HOME/.local/bin:$PATH

WORKDIR $HOME
RUN mkdir app
WORKDIR $HOME/app
COPY . $HOME/app

EXPOSE 8501
CMD streamlit run app.py \
    --server.headless true \
    --server.enableCORS false \
    --server.enableXsrfProtection false \
    --server.fileWatcherType none