From f4831de5166c44ed5cb619385092a4b8003bfd84 Mon Sep 17 00:00:00 2001 From: zouap Date: Fri, 3 Dec 2021 09:42:02 +0800 Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4=E4=BB=A3=E7=A0=81=E3=80=82?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: zouap --- routers/repo/ai_model_manage.go | 12 ++++++++---- routers/routes/routes.go | 1 + 2 files changed, 9 insertions(+), 4 deletions(-) diff --git a/routers/repo/ai_model_manage.go b/routers/repo/ai_model_manage.go index 1e5c9198c..5ce72762c 100644 --- a/routers/repo/ai_model_manage.go +++ b/routers/repo/ai_model_manage.go @@ -372,10 +372,14 @@ func ShowSingleModel(ctx *context.Context) { log.Info("get model file,size=" + fmt.Sprint(len(models))) } - ctx.Data["Dirs"] = models - ctx.Data["task"] = task - ctx.Data["ID"] = id - ctx.HTML(200, tplModelManageDownload) + if err != nil { + ctx.ServerError("QueryTrainJobList:", err) + } else { + mapInterface := make(map[string]interface{}) + mapInterface["task"] = task + mapInterface["models"] = models + ctx.JSON(http.StatusOK, mapInterface) + } } func ShowOneVersionOtherModel(ctx *context.Context) { diff --git a/routers/routes/routes.go b/routers/routes/routes.go index f30a52975..d0c8e0178 100755 --- a/routers/routes/routes.go +++ b/routers/routes/routes.go @@ -974,6 +974,7 @@ func RegisterRoutes(m *macaron.Macaron) { m.Put("/modify_model", repo.ModifyModelInfo) m.Get("/show_model", reqRepoCloudBrainReader, repo.ShowModelTemplate) m.Get("/show_model_info", reqRepoCloudBrainReader, repo.ShowModelInfo) + m.Get("/show_model_info_api", reqRepoCloudBrainReader, repo.ShowSingleModel) m.Get("/show_model_api", reqRepoCloudBrainReader, repo.ShowModelPageInfo) m.Get("/show_model_child_api", reqRepoCloudBrainReader, repo.ShowOneVersionOtherModel) m.Get("/query_train_job", reqRepoCloudBrainReader, repo.QueryTrainJobList)