tyang4 commited on
Commit
6e7a8e2
·
verified ·
1 Parent(s): 6143b93

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +4 -0
Dockerfile CHANGED
@@ -9,7 +9,11 @@ RUN apt-get update && apt-get install -y \
9
  git \
10
  && rm -rf /var/lib/apt/lists/*
11
 
 
 
 
12
  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')"
 
13
  COPY requirements.txt ./
14
  COPY src/ ./src/
15
 
 
9
  git \
10
  && rm -rf /var/lib/apt/lists/*
11
 
12
+ RUN pip install --no-cache-dir streamlit pandas transformers torch faiss-cpu scikit-learn neo4j pydeck requests
13
+
14
+
15
  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')"
16
+
17
  COPY requirements.txt ./
18
  COPY src/ ./src/
19