Rakshitjan's picture
Update Dockerfile
bddd65e verified
raw
history blame contribute delete
719 Bytes
FROM python:3.9-slim
# Set environment variables
ENV NLTK_DATA=/usr/share/nltk_data
ENV HF_HOME=/usr/share/huggingface
ENV TORCH_HOME=/usr/share/torch
# Create directories with proper permissions
RUN mkdir -p ${NLTK_DATA} ${HF_HOME} ${TORCH_HOME} && \
chown -R 1000:1000 /usr/share && \
apt-get update && \
apt-get install -y --no-install-recommends wget git && \
rm -rf /var/lib/apt/lists/*
WORKDIR /app
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt && \
python -m nltk.downloader punkt punkt_tab -d ${NLTK_DATA}
# Copy application code
COPY main.py .
# Run as non-root user
USER 1000
EXPOSE 7860
CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]