fix install deps in Dockerfile (#244)

Signed-off-by: chensuyue <[email protected]>
opea-project · Jun 26, 2024 · cf6043f · cf6043f
1 parent e8ebd94
commit cf6043f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/comps/dataprep/redis/langchain/docker/Dockerfile b/comps/dataprep/redis/langchain/docker/Dockerfile
@@ -23,7 +23,7 @@ USER user
 COPY comps /home/user/comps
 
 RUN pip install --no-cache-dir --upgrade pip setuptools && \
-    if [ ${ARCH} = "cpu" ]; then pip install torch --index-url https://download.pytorch.org/whl/cpu; fi && \
+    if [ ${ARCH} = "cpu" ]; then pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu; fi && \
     pip install --no-cache-dir -r /home/user/comps/dataprep/redis/langchain/requirements.txt
 
 ENV PYTHONPATH=$PYTHONPATH:/home/user