Siddhant commited on
Commit
01a5bd2
1 Parent(s): 01c0c82

torch package set to cpu

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. requirements.txt +5 -1
Dockerfile CHANGED
@@ -6,7 +6,7 @@ COPY . /app
6
  #RUN mkdir /app/data
7
  RUN pip install -r requirements.txt
8
 
9
- #EXPOSE 80
10
 
11
  RUN useradd -m -u 1000 user
12
 
 
6
  #RUN mkdir /app/data
7
  RUN pip install -r requirements.txt
8
 
9
+ #EXPOSE 7860
10
 
11
  RUN useradd -m -u 1000 user
12
 
requirements.txt CHANGED
@@ -4,4 +4,8 @@ python-dotenv==1.0.0
4
  uvicorn==0.22.0
5
  tiktoken==0.3.3
6
  weaviate-client==3.19.2
7
- sentence_transformers
 
 
 
 
 
4
  uvicorn==0.22.0
5
  tiktoken==0.3.3
6
  weaviate-client==3.19.2
7
+
8
+ --extra-index-url https://download.pytorch.org/whl/cpu
9
+ torch
10
+
11
+ sentence-transformers