更改推理结构

models/Image.go

@@ -37,6 +37,7 @@ type Image struct {
 	Progress       int       `json:"progress"`              // 任务进度(0-100)
 	Public         bool      `json:"public"`                // 是否公开
 	UserID         int       `json:"user_id"`               // 用户ID
+	ModelID        int       `json:"model_id"`              // 模型ID
 	CreatedAt      time.Time `json:"created_at" gorm:"autoCreateTime"`
 	UpdatedAt      time.Time `json:"updated_at" gorm:"autoUpdateTime"`
 }