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, }