fix(gui): do not persist loading flag

ssube · Jan 12, 2023 · dcfce81 · dcfce81
1 parent 25b64cc
commit dcfce81
Showing 1 changed file with 8 additions and 1 deletion.
diff --git 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),
   }));