diff --git a/routers/repo/aisafety.go b/routers/repo/aisafety.go
index 245894e45..49e4d7b6a 100644
--- a/routers/repo/aisafety.go
+++ b/routers/repo/aisafety.go
@@ -93,7 +93,7 @@ func GetAiSafetyTask(ctx *context.Context) {
 	syncAiSafetyTaskStatus(job)
 	job, err = models.GetCloudbrainByIDWithDeleted(ID)
 	job.BenchmarkType = "CV"
-	job.BenchmarkTypeName = "Classification"
+	job.BenchmarkTypeName = "图像分类"
 	ctx.JSON(200, job)
 }
 
diff --git a/routers/repo/cloudbrain.go b/routers/repo/cloudbrain.go
index b3069d00d..11e632620 100755
--- a/routers/repo/cloudbrain.go
+++ b/routers/repo/cloudbrain.go
@@ -2175,6 +2175,10 @@ func CloudBrainBenchmarkIndex(ctx *context.Context) {
 				}
 			}
 		}
+		if task.JobType == string(models.JobTypeModelSafety) {
+			ciTasks[i].BenchmarkType = "CV"
+			ciTasks[i].BenchmarkTypeName = "图像分类"
+		}
 	}
 
 	pager := context.NewPagination(int(count), setting.UI.IssuePagingNum, page, 5)