Update app.py

app.py

@@ -179,9 +179,15 @@ global pipeline
 global MultiResNetModel
 global causal_dit
 global controlnet
+global cur_style
 
 @spaces.GPU
 def load_ckpt():
+    global pipeline
+    global MultiResNetModel
+    global causal_dit
+    global controlnet
+    global cur_style
     weight_dtype = torch.float16
 
     block_out_channels = [128, 128, 256, 512, 512]
@@ -292,11 +298,14 @@ def load_ckpt():
     print('loaded pipeline')
 
 load_ckpt()
-
-global cur_style
 cur_style = 'line + shadow'
 @spaces.GPU
 def change_ckpt(style):
+    global pipeline
+    global MultiResNetModel
+    global causal_dit
+    global controlnet
+    global cur_style
     weight_dtype = torch.float16
 
     if style == 'line':
@@ -348,6 +357,11 @@ def process_multi_images(files):
 
 @spaces.GPU
 def extract_lines(image):
+    global pipeline
+    global MultiResNetModel
+    global causal_dit
+    global controlnet
+    global cur_style
     src = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2GRAY)
 
     rows = int(np.ceil(src.shape[0] / 16)) * 16
@@ -373,6 +387,11 @@ def extract_lines(image):
 
 @spaces.GPU
 def extract_line_image(query_image_, resolution):
+    global pipeline
+    global MultiResNetModel
+    global causal_dit
+    global controlnet
+    global cur_style
     tar_width, tar_height = resolution
     query_image = query_image_.resize((tar_width, tar_height))
     query_image = query_image.convert('L').convert('RGB')
@@ -418,6 +437,11 @@ def extract_sketch_line_image(query_image_, input_style):
 
 @spaces.GPU(duration=120)
 def colorize_image(extracted_line, reference_images, resolution, seed, num_inference_steps, top_k, hint_mask=None, hint_color=None, query_image_origin=None, extracted_image_ori=None):
+    global pipeline
+    global MultiResNetModel
+    global causal_dit
+    global controlnet
+    global cur_style
     if extracted_line is None:
         gr.Info("Please preprocess the image first")
         raise ValueError("Please preprocess the image first")