null

models/Model.go

@@ -46,26 +46,3 @@ func (model *Model) Train() (err error) {
 	}
 	return
 }
-
-//func (model *Model) SendToTrain() error {
-//	db, err := configs.GetDB()
-//	if err != nil {
-//		log.Println(err)
-//		return err
-//	}
-//	defer db.Close()
-//	stmt, err := db.Prepare("UPDATE models SET status = ?, progress = ?, updated_at = ? WHERE id = ?")
-//	if err != nil {
-//		log.Println(err)
-//		return err
-//	}
-//	defer stmt.Close()
-//	_, err = stmt.Exec(model.Status, model.Progress, model.UpdatedAt, model.ID)
-//	if err != nil {
-//		log.Println(err)
-//		return err
-//	}
-//	// TODO: 創建一個新線程管理訓練任務
-//	// 將任務放入隊列中, 將自動回調更新任務狀態
-//	return nil
-//}

routers/servers.go

@@ -31,6 +31,7 @@ func ServersGet(w http.ResponseWriter, r *http.Request) {
 		// 讀取模型信息
 		resp, err = http.Get(fmt.Sprintf("http://%s:%d/sdapi/v1/sd-models", server.IP, server.Port))
 		if err != nil || resp.StatusCode != http.StatusOK {
+			server.Models = []map[string]interface{}{}
 		} else {
 			var models []map[string]interface{}
 			body, _ := ioutil.ReadAll(resp.Body)