Update app.py

app.py

@@ -8,10 +8,23 @@ import logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger("CodeSecurityAPI")
 
-# 强制设置缓存路径（解决权限问题）
+# 强制设置缓存路径
 os.environ["HF_HOME"] = "/app/.cache/huggingface"
 
-# 加载模型
+app = FastAPI()
+
+# === 新增根路径响应 ===
+@app.get("/")
+async def read_root():
+    return {
+        "message": "欢迎使用代码安全检测API",
+        "endpoints": {
+            "detect": "POST /detect",
+            "health": "GET /health"
+        }
+    }
+
+# === 加载模型（必须放在FastAPI实例之后） ===
 try:
     logger.info("Loading model...")
     model = AutoModelForSequenceClassification.from_pretrained(
@@ -27,26 +40,18 @@ except Exception as e:
     logger.error(f"Model load failed: {str(e)}")
     raise RuntimeError("模型加载失败，请检查网络连接或模型路径")
 
-app = FastAPI()
-
 @app.post("/detect")
 async def detect(code: str):
     try:
-        # 输入处理（限制长度）
-        code = code[:2000]  # 截断超长输入
-        
-        # 模型推理
+        code = code[:2000]
         inputs = tokenizer(code, return_tensors="pt", truncation=True, max_length=512)
         with torch.no_grad():
             outputs = model(**inputs)
-        
-        # 解析结果
         label_id = outputs.logits.argmax().item()
         return {
             "label": model.config.id2label[label_id],
             "score": outputs.logits.softmax(dim=-1)[0][label_id].item()
         }
-        
     except Exception as e:
         return {"error": str(e)}