From 4fe977eebbb5f0b694671f14e1505f4b31874202 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=A1=9C=E8=8F=AF?= <huan0016@gmail.com>
Date: Tue, 20 Jun 2023 19:57:39 +0800
Subject: [PATCH] =?UTF-8?q?=E5=85=81=E8=AE=B8ws=E8=B7=A8=E5=9F=9F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 routers/images.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/routers/images.go b/routers/images.go
index 8854228..bd6c23f 100644
--- a/routers/images.go
+++ b/routers/images.go
@@ -30,6 +30,9 @@ func ImagesGet(w http.ResponseWriter, r *http.Request) {
 	// websocket 推理图像
 	if r.Header.Get("Upgrade") == "websocket" {
 		upgrader := websocket.Upgrader{}
+		upgrader.CheckOrigin = func(r *http.Request) bool {
+			return true
+		}
 		conn, err := upgrader.Upgrade(w, r, nil)
 		if err != nil {
 			log.Println(err)
@@ -147,7 +150,6 @@ func ImagesPost(w http.ResponseWriter, r *http.Request) {
 
 			w.Header().Set("Content-Type", "application/json; charset=utf-8")
 			json.NewEncoder(w).Encode(image_list)
-			//w.Write(utils.ToJSON({"task": task, "list": image_list}))
 			return
 		}