diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000..4d8ef96 --- /dev/null +++ b/Dockerfile @@ -0,0 +1,22 @@ +FROM python:3.9 +ENV TZ=Asia/Shanghai +RUN apt update +RUN apt install -y vim +RUN apt install -y libgl1-mesa-glx +COPY ./requirements.txt /requirements.txt +RUN pip install --upgrade pip +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 + +WORKDIR /app +COPY . . +ENV FLASK_APP=manage.py +LABEL maintainer="zchengrong@yeah.net" \ + description="My Python 3.9 - trinity mixi " \ + version="1.0" \ + name="trinity_mixi" + + +CMD ["gunicorn", "-c", "gunicorn_config.py", "manage:app"] \ No newline at end of file diff --git a/app/service/outfit_matcher/dataset.py b/app/service/outfit_matcher/dataset.py index 21766ff..e440c50 100644 --- a/app/service/outfit_matcher/dataset.py +++ b/app/service/outfit_matcher/dataset.py @@ -11,12 +11,12 @@ def sim_score(a, b): class FashionDataset(object): - # fashion_categories = json.load(open(r"config/fashion_categories.json", "r")) - # fashion_categories_mapping = json.load(open(r"config/fashion_category_mapping.json", "r")) + fashion_categories = json.load(open(r"config/fashion_categories.json", "r")) + fashion_categories_mapping = json.load(open(r"config/fashion_category_mapping.json", "r")) # service debug - fashion_categories = json.load(open(r"service/outfit_matcher/config/fashion_categories.json", "r")) - fashion_categories_mapping = json.load(open(r"service/outfit_matcher/config/fashion_category_mapping.json", "r")) + # fashion_categories = json.load(open(r"service/outfit_matcher/config/fashion_categories.json", "r")) + # fashion_categories_mapping = json.load(open(r"service/outfit_matcher/config/fashion_category_mapping.json", "r")) def __init__(self, item_metadata): self.item_metadat = item_metadata