|
@@ -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"
|