update

Dockerfile

@@ -13,15 +13,7 @@ RUN chown -R appuser:appuser /app
 RUN chmod -R 755 /app
 
 RUN mkdir -p /app/.cache
-# RUN mkdir -p /app/.cache/models--BAAI--bge-large-zh-v1.5
-# RUN mkdir -p /app/.cache/models--BAAI--bge-large-zh-v1.5/refs
-# RUN mkdir -p /app/.chach
-# RUN chown -R appuser:appuser /app/.chach
-# RUN chmod -R 755 /app/.chach
-
-# RUN mkdir -p /app/.chach/models--BAAI--bge-large-zh-v1.5 
-# RUN chown -R appuser:appuser /app/.chach
-# RUN chmod -R 755 /app/.chach
+
 # 设置工作目录
 WORKDIR /app
 
@@ -44,10 +36,5 @@ RUN chmod -R 777 /app/.cache
 # 切换到 appuser 用户
 USER appuser
 
-# RUN chown -R appuser:appuser /app/.cache/models--BAAI--bge-large-zh-v1.5
-# RUN chmod -R 775 /app/.cache/models--BAAI--bge-large-zh-v1.5
-# RUN chown -R appuser:appuser /app/.cache/models--BAAI--bge-large-zh-v1.5/refs
-# RUN chmod -R 775 /app/.cache/models--BAAI--bge-large-zh-v1.5/refs
-
 # 指定容器启动时运行的命令
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -1,13 +1,19 @@
-from fastapi import FastAPI, Request, HTTPException
+from fastapi import FastAPI, Request, HTTPException, Depends, Header
 from pydantic import BaseModel, Field
 from sentence_transformers import SentenceTransformer
 import numpy as np
-import logging
+import logging, os
 
 # 设置日志记录
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+# 定义依赖项来校验 Authorization
+async def check_authorization(authorization: str = Header(..., alias="Authorization")):
+    if authorization != os.environ.get("AUTHORIZATION"):
+        raise HTTPException(status_code=401, detail="Unauthorized access")
+    return authorization
+
 app = FastAPI()
 
 try: