Kaynağa Gözat

allow all cors

Daniel Gatis 3 yıl önce
ebeveyn
işleme
f339a11036
1 değiştirilmiş dosya ile 9 ekleme ve 0 silme
  1. 9 0
      rembg/cli.py

+ 9 - 0
rembg/cli.py

@@ -10,6 +10,7 @@ import filetype
 import uvicorn
 import uvicorn
 from asyncer import asyncify
 from asyncer import asyncify
 from fastapi import Depends, FastAPI, File, Query
 from fastapi import Depends, FastAPI, File, Query
+from fastapi.middleware.cors import CORSMiddleware
 from starlette.responses import Response
 from starlette.responses import Response
 from tqdm import tqdm
 from tqdm import tqdm
 from watchdog.events import FileSystemEvent, FileSystemEventHandler
 from watchdog.events import FileSystemEvent, FileSystemEventHandler
@@ -270,6 +271,14 @@ def s(port: int, log_level: str) -> None:
         openapi_tags=tags_metadata,
         openapi_tags=tags_metadata,
     )
     )
 
 
+    app.add_middleware(
+        CORSMiddleware,
+        allow_credentials=True,
+        allow_origins=["*"],
+        allow_methods=["*"],
+        allow_headers=["*"],
+    )
+
     class ModelType(str, Enum):
     class ModelType(str, Enum):
         u2net = "u2net"
         u2net = "u2net"
         u2netp = "u2netp"
         u2netp = "u2netp"