From 36aa83999193146172018d6861ce4df1fa1220c5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=A1=9C=E8=8F=AF?= <huan0016@gmail.com>
Date: Fri, 26 May 2023 04:19:29 +0800
Subject: [PATCH] null

---
 models/Model.go    | 23 -----------------------
 routers/servers.go |  1 +
 2 files changed, 1 insertion(+), 23 deletions(-)

diff --git a/models/Model.go b/models/Model.go
index a4787a3..455df18 100644
--- a/models/Model.go
+++ b/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
-//}
diff --git a/routers/servers.go b/routers/servers.go
index dbb2b22..09044b8 100644
--- a/routers/servers.go
+++ b/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)