fixed dockerfile for ai processing

nlp4se · Feb 27, 2025 · b849ca4 · b849ca4
1 parent d4cd5ee
commit b849ca4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -6,6 +6,8 @@ COPY Pipfile Pipfile.lock /wsgi/
 
 RUN pip install pipenv && pipenv install --deploy --ignore-pipfile
 
+RUN pipenv run pip install torch --index-url https://download.pytorch.org/whl/cpu
+
 RUN pipenv run python -m spacy download en_core_web_sm
 
 COPY . /wsgi