From 2f342a45eadb15082666414728f77ed08f7fc31b Mon Sep 17 00:00:00 2001
From: ychao_1983 <ychao_1983@sina.com>
Date: Fri, 16 Dec 2022 10:42:43 +0800
Subject: [PATCH] =?UTF-8?q?npu=E8=B0=83=E8=AF=95=E6=A8=A1=E5=9E=8B?=
 =?UTF-8?q?=E5=9C=B0=E5=9D=80=E6=94=B9=E4=B8=BAhttps=E6=A0=BC=E5=BC=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

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

diff --git a/routers/repo/modelarts.go b/routers/repo/modelarts.go
index 828590564..19b61cb44 100755
--- a/routers/repo/modelarts.go
+++ b/routers/repo/modelarts.go
@@ -346,9 +346,12 @@ func NotebookShow(ctx *context.Context) {
 }
 
 func GetModelDownload(task *models.Cloudbrain) models.ModelDownload {
+	index := strings.Index(task.PreTrainModelUrl, "/")
+	key := task.PreTrainModelUrl[index+1:] + task.CkptName
+	url, _ := storage.GetObsCreateSignedUrlByBucketAndKey(setting.Bucket, key)
 	modelDownload := models.ModelDownload{
 		Name:         task.CkptName,
-		DownloadLink: "s3://" + task.PreTrainModelUrl + task.CkptName,
+		DownloadLink: url,
 		IsDelete:     false,
 	}