Update app.py

app.py

@@ -174,14 +174,16 @@ image_encoder = CLIPVisionModelWithProjection.from_pretrained(os.path.join(model
 # model_sketch = create_model_sketch('default').to('cuda')      # create a model given opt.model and other options
 # model_sketch.eval()
 
-
 global pipeline
 global MultiResNetModel
 global cur_style
-cur_style = 'line + shadow'
 
 @spaces.GPU
 def load_ckpt():
+    global pipeline
+    global MultiResNetModel
+    global cur_style
+    cur_style = 'line + shadow'
     weight_dtype = torch.float16
 
     block_out_channels = [128, 128, 256, 512, 512]
@@ -291,10 +293,8 @@ def load_ckpt():
 
     print('loaded pipeline')
 
-    return pipeline, MultiResNetModel
-
 
-pipeline, MultiResNetModel = load_ckpt()
+load_ckpt()
 
 @spaces.GPU
 def change_ckpt(style):
@@ -311,6 +311,10 @@ def change_ckpt(style):
     else:
         raise ValueError("Invalid style: {}".format(style))
 
+    global pipeline
+    global MultiResNetModel
+    global cur_style
+
     cur_style = style
 
     MultiResNetModel.load_state_dict(torch.load(MultiResNetModel_path, map_location='cpu'), strict=True)
@@ -349,6 +353,7 @@ def process_multi_images(files):
 
 @spaces.GPU
 def extract_lines(image):
+    global line_model
     src = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2GRAY)
 
     rows = int(np.ceil(src.shape[0] / 16)) * 16
@@ -384,6 +389,7 @@ def extract_line_image(query_image_, resolution):
 
 @spaces.GPU
 def extract_sketch_line_image(query_image_, input_style):
+    global cur_style
     if input_style != cur_style:
         change_ckpt(input_style)
 
@@ -425,6 +431,10 @@ def colorize_image(extracted_line, reference_images, resolution, seed, num_infer
     reference_images = process_multi_images(reference_images)
     fix_random_seeds(seed)
 
+    global pipeline
+    global MultiResNetModel
+    global cur_style
+
     tar_width, tar_height = resolution
 
     gr.Info("Image retrieval in progress...")