diff --git a/routers/repo/aisafety.go b/routers/repo/aisafety.go
index 1320ad0db..9bb4b9dd4 100644
--- a/routers/repo/aisafety.go
+++ b/routers/repo/aisafety.go
@@ -978,6 +978,12 @@ func modelSafetyNewDataPrepare(ctx *context.Context) error {
 		ctx.Data["CombatDataSetName"] = setting.ModelSafetyTest.GPUCombatDataSetName
 		ctx.Data["CombatDataSetUUID"] = setting.ModelSafetyTest.GPUCombatDataSetUUID
 		prepareCloudbrainOneSpecs(ctx)
+
+		queuesDetail, _ := cloudbrain.GetQueuesDetail()
+		if queuesDetail != nil {
+			ctx.Data["QueuesDetail"] = queuesDetail
+		}
+
 	} else {
 		ctx.Data["engine_id"] = ctx.QueryInt("engine_id")
 		ctx.Data["pool_id"] = ctx.Query("pool_id")
@@ -1002,6 +1008,9 @@ func modelSafetyNewDataPrepare(ctx *context.Context) error {
 		}
 		ctx.Data["engine_versions"] = versionInfos.Version
 		prepareCloudbrainTwoInferenceSpecs(ctx)
+
+		waitCount := cloudbrain.GetWaitingCloudbrainCount(models.TypeCloudBrainTwo, "")
+		ctx.Data["WaitCount"] = waitCount
 	}
 
 	return nil