pfe_site / Dockerfile
YsnHdn's picture
Refactor : Using DistilBert instead of Bert
3782af5
raw
history blame
904 Bytes
# read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
# you will also find guides on how best to write your Dockerfile
FROM python:3.9
WORKDIR /code
# Copy requirements.txt separately to leverage Docker layer caching
COPY ./requirements.txt /code/requirements.txt
# Install dependencies
RUN pip install --no-cache-dir --upgrade -r requirements.txt
# Set the HF_HOME environment variable
ENV HF_HOME /code/.cache/huggingface
# Create cache directory with appropriate permissions
RUN mkdir -p $HF_HOME && chmod -R 777 $HF_HOME
# Copy the model files into the image
COPY ./distilBert /code/distilBert
# Copy the rest of the application files
COPY . .
# Grant write permissions to the directory where your application saves files
RUN mkdir -p /code/static/uploads && chmod -R 777 /code/static/uploads
# Command to run your application
CMD ["gunicorn", "-b", "0.0.0.0:7860", "app:app"]