preview

models/Model.go

@@ -32,7 +32,7 @@ type Model struct {
 	ModelPath       string    `json:"model_path"`               // 模型路徑(實際存放在服務器上的路徑)
 	Status          string    `json:"status" default:"initial"` // (initial|ready|waiting|running|success|error)
 	Progress        int       `json:"progress"`                 // (0-100)
-	Image           string    `json:"image"`                    // 封面圖片實際地址
+	Preview         string    `json:"preview"`                  // 模型預覽圖片
 	Hash            string    `json:"hash"`                     // 模型哈希值(sha256)
 	Epochs          int       `json:"epochs"`                   // 訓練步數
 	LearningRate    float32   `json:"learning_rate"`            // 學習率(0.000005)

routers/models.go

@@ -248,8 +248,8 @@ func ModelItemPatch(w http.ResponseWriter, r *http.Request) {
 			//go model.Train()
 		}
 	}
-	if model_new.Image != "" && model_new.Image != model.Image {
-		model.Image = model_new.Image
+	if model_new.Preview != "" && model_new.Preview != model.Preview {
+		model.Preview = model_new.Preview
 	}
 	if model_new.TriggerWords != "" && model_new.TriggerWords != model.TriggerWords {
 		model.TriggerWords = model_new.TriggerWords