From dcfce81fedffe7e67805f1822f0aec39687fedd4 Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Wed, 11 Jan 2023 21:45:47 -0600
Subject: [PATCH] fix(gui): do not persist loading flag

---
 gui/src/main.tsx | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/gui/src/main.tsx b/gui/src/main.tsx
index 1f8828c7..292e7a1a 100644
--- a/gui/src/main.tsx
+++ b/gui/src/main.tsx
@@ -47,7 +47,7 @@ export async function main() {
   merge(params, config.params);
 
   const defaults = paramsFromConfig(params);
-  const state = createStore<OnnxState, [['zustand/persist', never]]>(persist((set) => ({
+  const state = createStore<OnnxState, [['zustand/persist', OnnxState]]>(persist((set) => ({
     defaults,
     history: {
       images: [],
@@ -170,6 +170,13 @@ export async function main() {
     },
   }), {
     name: 'onnx-web',
+    partialize: (oldState) => ({
+      ...oldState,
+      history: {
+        ...oldState.history,
+        loading: false,
+      },
+    }),
     storage: createJSONStorage(() => localStorage),
   }));