diff --git a/routers/repo/aisafety.go b/routers/repo/aisafety.go
index 30cb50c79..1eaa2669f 100644
--- a/routers/repo/aisafety.go
+++ b/routers/repo/aisafety.go
@@ -509,6 +509,8 @@ func AiSafetyCreateForGetGPU(ctx *context.Context) {
 	t := time.Now()
 	ctx.Data["PageIsCloudBrain"] = true
 	ctx.Data["IsCreate"] = true
+	ctx.Data["type"] = models.TypeCloudBrainOne
+	ctx.Data["compute_resource"] = models.GPUResource
 	ctx.Data["datasetType"] = models.TypeCloudBrainOne
 	ctx.Data["BaseDataSetName"] = setting.ModelSafetyTest.BaseDataSetName
 	ctx.Data["BaseDataSetUUID"] = setting.ModelSafetyTest.BaseDataSetUUID
@@ -565,6 +567,8 @@ func AiSafetyCreateForGetNPU(ctx *context.Context) {
 	t := time.Now()
 	ctx.Data["PageIsCloudBrain"] = true
 	ctx.Data["IsCreate"] = true
+	ctx.Data["type"] = models.TypeCloudBrainTwo
+	ctx.Data["compute_resource"] = models.GPUResource
 	var displayJobName = jobNamePrefixValid(cutString(ctx.User.Name, 5)) + t.Format("2006010215") + strconv.Itoa(int(t.Unix()))[5:]
 	ctx.Data["display_job_name"] = displayJobName
 	ctx.Data["datasetType"] = models.TypeCloudBrainTwo