Update app.py

app.py

@@ -146,7 +146,7 @@ opt.colors = [
     # '#92C6EC',
     # '#FECAC0',
 ]
-opt.excluded_keys = ['inpainting_mode', 'is_runing', 'active_palettes', 'current_palette', 'model']
+opt.excluded_keys = ['inpainting_mode', 'is_running', 'active_palettes', 'current_palette', 'model']
 
 
 ### Event handlers
@@ -292,13 +292,15 @@ def change_neg_prompt(state, neg_prompt):
 
 
 def import_state(state, json_text):
-    current_palette = state.current_palette
-    # active_palettes = state.active_palettes
+    prev_state_dict = {k: v in vars(state).items() if k in opt.excluded_keys}
     state_dict = json.loads(json_text)
     for k in opt.excluded_keys:
         if k in state_dict:
             del state_dict[k]
+    state_dict.update(prev_state_dict)
     state = argparse.Namespace(**state_dict)
+
+    current_palette = state.current_palette
     state.active_palettes = opt.max_palettes
     return [state] + [
         gr.update(value=v, visible=True) for v in state.prompt_names
@@ -317,9 +319,6 @@ def import_state(state, json_text):
 
 ### Main worker
 
-def generate():
-    return model()
-
 
 def register(state, drawpad):
     seed_everything(state.seed if state.seed >=0 else np.random.randint(2147483647))
@@ -395,12 +394,12 @@ def run(state, drawpad):
 
     tic = time.time()
     while True:
-        yield [state, generate()]
+        yield [state, state.model()]
         toc = time.time()
         tdelta = toc - tic
         if tdelta > opt.run_time:
             state.is_running = False
-            return [state, generate()]
+            return [state, state.model()]
 
 
 def hide_element():