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