dataset 上传文件

2023-07-11 14:37:03 +08:00
parent fd8335613a
commit 050637dcd8
2 changed files with 84 additions and 0 deletions
--- a/models/Model.go
+++ b/models/Model.go
@@ -63,6 +63,12 @@ func (model *Model) Load() {
 	configs.ORMDB().First(&model)
 }

+// 从数据库加载指定的模型
+func ModelLoad(id int) (model Model, err error) {
+	err = configs.ORMDB().First(&model, id).Error
+	return
+}
+
 // 推理模型
 func (model *Model) Inference(image_list []Image, callback func(Image)) {
 	var server Server