Update app.py

app.py

@@ -308,7 +308,6 @@ def change_ckpt(style):
     global MultiResNetModel
     global cur_style
 
-    cur_style = style
 
     MultiResNetModel.load_state_dict(torch.load(MultiResNetModel_path, map_location='cpu'), strict=True)
     MultiResNetModel.to('cuda', dtype=weight_dtype)
@@ -427,6 +426,7 @@ def colorize_image(input_style, extracted_line, reference_images, resolution, se
     if input_style != cur_style:
         gr.Info("Loading the model...")
         change_ckpt(input_style)
+        cur_style = input_style
 
     tar_width, tar_height = resolution