Function predict

python_server/server.py:127–142

(req: PredictRequest)

@app.post("/predict", response_model=PredictResponse)
@app.post("/predict_logits", response_model=PredictResponse) # 别名路由，防备队友写错路径
async def predict(req: PredictRequest):
    if model is None: raise HTTPException(500, "Model not ready")
    if len(req.image) != FLATTENED_SIZE: raise HTTPException(400, "Shape Error")

    try:
        input_tensor = torch.tensor(req.image).float().to(DEVICE).view(1, 3, 32, 32)
        
        with torch.no_grad(): # 必加：防止 OOM
            output = model(input_tensor)
            pred_label = output.argmax(dim=1).item()
            logits_list = output.cpu().squeeze().tolist()
            
            return PredictResponse(label=pred_label, logits=logits_list)
            
    except Exception as e:
        raise HTTPException(500, str(e))

# 接口 2: 批量预测 (支撑 5x11 并发)
@app.post("/predict_batch", response_model=BatchPredictResponse)