diff --git a/routers/repo/ai_model_manage.go b/routers/repo/ai_model_manage.go index 7ef9045f5..a71203b40 100644 --- a/routers/repo/ai_model_manage.go +++ b/routers/repo/ai_model_manage.go @@ -447,13 +447,14 @@ func DeleteModelFile(ctx *context.Context) { err := storage.Attachments.DeleteDir(model.Path[len(bucketName)+1:] + fileName) if err != nil { log.Info("Failed to delete model. id=" + id) - } else { re := map[string]string{ "code": "-1", } re["msg"] = err.Error() ctx.JSON(200, re) return + } else { + UpdateModelSize(id) } } } else if model.Type == models.TypeCloudBrainTwo { @@ -462,13 +463,14 @@ func DeleteModelFile(ctx *context.Context) { err := storage.ObsRemoveObject(setting.Bucket, model.Path[len(setting.Bucket)+1:]+fileName) if err != nil { log.Info("Failed to delete model. id=" + id) - } else { re := map[string]string{ "code": "-1", } re["msg"] = err.Error() ctx.JSON(200, re) return + } else { + UpdateModelSize(id) } } }