Update server.py

接口服务允许跨域请求,接入其他服务,比如ollama或者其他大模型服务
This commit is contained in:
刘悦
2024-07-11 18:11:03 +08:00
committed by GitHub
parent c2f9254006
commit a8b46cf810

View File

@@ -10,6 +10,7 @@ import sys
import io,time import io,time
from fastapi import FastAPI, Response, File, UploadFile, Form from fastapi import FastAPI, Response, File, UploadFile, Form
from fastapi.responses import HTMLResponse from fastapi.responses import HTMLResponse
from fastapi.middleware.cors import CORSMiddleware #引入 CORS中间件模块
from contextlib import asynccontextmanager from contextlib import asynccontextmanager
ROOT_DIR = os.path.dirname(os.path.abspath(__file__)) ROOT_DIR = os.path.dirname(os.path.abspath(__file__))
sys.path.append('{}/../../..'.format(ROOT_DIR)) sys.path.append('{}/../../..'.format(ROOT_DIR))
@@ -39,6 +40,15 @@ async def lifespan(app: FastAPI):
app = FastAPI(lifespan=lifespan) app = FastAPI(lifespan=lifespan)
#设置允许访问的域名
origins = ["*"] #"*",即为所有,也可以改为允许的特定ip。
app.add_middleware(
CORSMiddleware,
allow_origins=origins, #设置允许的origins来源
allow_credentials=True,
allow_methods=["*"], # 设置允许跨域的http方法比如 get、post、put等。
allow_headers=["*"]) #允许跨域的headers可以用来鉴别来源等作用。
def buildResponse(output): def buildResponse(output):
buffer = io.BytesIO() buffer = io.BytesIO()
torchaudio.save(buffer, output, 22050, format="wav") torchaudio.save(buffer, output, 22050, format="wav")