diff --git a/Dockerfile b/Dockerfile index b3b6b71..3b8db2c 100644 --- a/Dockerfile +++ b/Dockerfile @@ -16,6 +16,7 @@ WORKDIR /app # Установка зависимостей RUN pip install --upgrade pip RUN pip install --no-cache-dir -r requirements.txt +RUN pip install torch --index-url https://download.pytorch.org/whl/cpu # Копирование исходного кода COPY app /app diff --git a/app/requirements.txt b/app/requirements.txt index 081d857..72b0b7a 100644 --- a/app/requirements.txt +++ b/app/requirements.txt @@ -1,7 +1,6 @@ fastapi uvicorn[standard] transformers -torch --index-url https://download.pytorch.org/whl/cpu gunicorn aioredis==2.0.1 celery==5.3.0