diff --git a/tools/summarizer/Dockerfile b/tools/summarizer/Dockerfile index fe1c1e3..f3f3a6e 100644 --- a/tools/summarizer/Dockerfile +++ b/tools/summarizer/Dockerfile @@ -1,4 +1,4 @@ -FROM python:3.11-slim +FROM python:3.11 ENV PIP_NO_CACHE_DIR=1 \ PIP_DISABLE_PIP_VERSION_CHECK=1 \ PYTHONUNBUFFERED=1 \ @@ -6,9 +6,8 @@ ENV PIP_NO_CACHE_DIR=1 \ HF_HUB_DISABLE_TELEMETRY=1 WORKDIR /app COPY requirements.txt ./ -RUN pip install --upgrade pip setuptools wheel \ - && pip install --extra-index-url https://download.pytorch.org/whl/cpu torch==2.5.1 \ - && pip install -r requirements.txt +RUN python -m pip install --upgrade pip setuptools wheel \ + && python -m pip install --extra-index-url https://download.pytorch.org/whl/cpu -r requirements.txt COPY . . EXPOSE 8001 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8001"]