ws更新触发

routers/images.go

@@ -154,11 +154,6 @@ func ImagesPost(w http.ResponseWriter, r *http.Request) {
 				return
 			}
 
-			// 推理图像
-			go model.Inference([]models.Image{}, func() {
-				log.Println("推理完成")
-			})
-
 			// 直接创建一组图片
 			var image_list []models.Image
 			var task string = uuid.New().String()
@@ -177,8 +172,10 @@ func ImagesPost(w http.ResponseWriter, r *http.Request) {
 				image_list = append(image_list, image)
 			}
 
-			go model.Inference(image_list, func() {
-				images_websocket_manager.NotifyTaskChange(task, image_list)
+			// 推理图像
+			go model.Inference([]models.Image{}, func(img models.Image) {
+				log.Println("推理完成")
+				images_websocket_manager.NotifyTaskChange(task, img)
 			})
 
 			// 存储图片信息到数据库