diff --git a/Dockerfile b/Dockerfile index d1ddc5d..286d1a2 100644 --- a/Dockerfile +++ b/Dockerfile @@ -12,7 +12,7 @@ RUN pip install -r requirements.txt RUN pip install gunicorn RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 RUN pip install mmcv==1.4.2 -f https://download.openmmlab.com/mmcv/dist/cu117/torch1.13/index.html -RUN + WORKDIR /app COPY . . RUN mkdir -p app/logs diff --git a/app/api/api_process_lookbooks.py b/app/api/api_process_lookbooks.py index b019e0c..fbda8c8 100644 --- a/app/api/api_process_lookbooks.py +++ b/app/api/api_process_lookbooks.py @@ -3,8 +3,9 @@ import os import shutil from typing import List +import aiofiles import tqdm -from fastapi import UploadFile, File, APIRouter, BackgroundTasks +from fastapi import UploadFile, File, APIRouter, BackgroundTasks, Form from app.service.lookbooks.service import create_image_batch_requests, process_lookbook_task # 引入服务逻辑 logger = logging.getLogger() diff --git a/requirements.txt b/requirements.txt index 1ca8927..8ce3bab 100644 Binary files a/requirements.txt and b/requirements.txt differ