Add docker file

.dockerignore

@@ -0,0 +1,3 @@
+__pycache__
+.checkpoint
+Dockerfile

Dockerfile

@@ -0,0 +1,17 @@
+FROM python:3.11
+
+WORKDIR /src
+
+ADD ./ /src
+
+RUN pip install --upgrade pip
+RUN pip install --no-cache-dir -r requirements.txt
+
+
+# Download Model
+RUN python3 -c 'from app.cloth_segmentation.model import load_seg_model; load_seg_model(".checkpoint/model.pth")'
+
+EXPOSE 80
+
+
+CMD ["uvicorn", "app.server:app", "--proxy-headers", "--host", "0.0.0.0", "--port", "80"]

api/index.py

@@ -1 +0,0 @@
-from app.server import app

app/server.py

@@ -1,14 +1,12 @@
-from typing import Optional
-from fastapi import FastAPI, UploadFile
-from contextlib import asynccontextmanager
-
-from pydantic.dataclasses import dataclass
-from app.cloth_segmentation.model import segment, Mode, load_seg_model, Result, U2NET
-from PIL import Image
-
-from base64 import b64encode
 from io import BytesIO
-
+from base64 import b64encode
+from PIL import Image
+from app.cloth_segmentation.model import segment, Mode, load_seg_model, Result, U2NET
+from pydantic.dataclasses import dataclass
+from contextlib import asynccontextmanager
+from fastapi.middleware.cors import CORSMiddleware
+from fastapi import FastAPI, UploadFile
+from typing import Optional
 
 # === DTOs ==
 
@@ -45,6 +43,13 @@ async def lifespan(app: FastAPI):
 
 
 app = FastAPI(lifespan=lifespan)
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=['*'],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
 # === Routes
 
 

requirements.txt

@@ -1,8 +1,9 @@
 uvicorn==0.23.2
 fastapi==0.104.1
+python-multipart==0.0.6
 pydantic==2.4.2
 
 Pillow==9.5.0
 numpy==1.24.3
-torch==2.1.0
-torchvision==0.16.0
+torch==2.0.*
+torchvision