From b585070baf2cf31f66ad7b3610213896b39b4902 Mon Sep 17 00:00:00 2001
From: liuzx <liuzx@pcl.ac.cn>
Date: Wed, 5 Jan 2022 09:56:33 +0800
Subject: [PATCH] update

---
 routers/repo/modelarts.go | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/routers/repo/modelarts.go b/routers/repo/modelarts.go
index 554bc4c41..7562478ff 100755
--- a/routers/repo/modelarts.go
+++ b/routers/repo/modelarts.go
@@ -1754,6 +1754,19 @@ func InferenceJobIndex(ctx *context.Context) {
 		tasks[i].ComputeResource = models.NPUResource
 	}
 
+	repoId := ctx.Repo.Repository.ID
+	Type := -1
+	_, model_count, _ := models.QueryModel(&models.AiModelQueryOptions{
+		ListOptions: models.ListOptions{
+			Page:     1,
+			PageSize: 2,
+		},
+		RepoID: repoId,
+		Type:   Type,
+		New:    MODEL_LATEST,
+	})
+	ctx.Data["MODEL_COUNT"] = model_count
+
 	pager := context.NewPagination(int(count), setting.UI.IssuePagingNum, page, 5)
 	pager.SetDefaultParams(ctx)
 	ctx.Data["Page"] = pager