File size: 1,272 Bytes
13e4952 1f2240e 80ab0bd 1f2240e 80ab0bd 71238f5 0a0925a 80ab0bd 8fede8e e0e8b60 80ab0bd befc054 80ab0bd befc054 1f2240e ea51847 6e7a8e2 30be17c 65f63ec 1f2240e befc054 1f2240e 22cc8b5 80ab0bd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
FROM python:3.10-slim
RUN apt-get update && apt-get install -y \
bash \
build-essential \
curl \
software-properties-common \
git\
git-lfs \
wget \
sudo \
procps \
htop \
vim \
neovim && \
rm -rf /var/lib/apt/lists/*
RUN useradd -m -u 1000 user
USER user
ENV PATH="/home/user/.local/bin:$PATH"
WORKDIR /app
COPY --chown=user ./requirements.txt requirements.txt
RUN pip install --no-cache-dir streamlit pandas transformers torch faiss-cpu scikit-learn neo4j pydeck requests folium streamlit-folium
#RUN python -c "from transformers import AutoTokenizer, AutoModel; AutoTokenizer.from_pretrained('sentence-transformers/all-MiniLM-L6-v2'); AutoModel.from_pretrained('sentence-transformers/all-MiniLM-L6-v2')"
COPY model/ ./model/
COPY requirements.txt ./
COPY src/ ./src/
RUN pip install -r requirements.txt
COPY --chown=user . /app
EXPOSE 8501
HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
ENV NEO4J_URI=$NEO4J_URI
ENV NEO4J_USERNAME=$NEO4J_USERNAME
ENV NEO4J_PASSWORD=$NEO4J_PASSWORD
ENV OPENAI_API_KEY=$openai_api_key
ENV STREAMLIT_HOME="/tmp/.streamlit"
ENV TRANSFORMERS_CACHE="/tmp/hf_cache"
CMD ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"] |