Update app.py

app.py

@@ -177,13 +177,11 @@ image_encoder = CLIPVisionModelWithProjection.from_pretrained(os.path.join(model
 
 global pipeline
 global MultiResNetModel
+global causal_dit
+global controlnet
 
 @spaces.GPU
 def load_ckpt():
-    global pipeline
-    global MultiResNetModel
-    global causal_dit
-    global controlnet
     weight_dtype = torch.float16
 
     block_out_channels = [128, 128, 256, 512, 512]
@@ -297,11 +295,6 @@ global cur_style
 cur_style = 'line + shadow'
 @spaces.GPU
 def change_ckpt(style):
-    global pipeline
-    global MultiResNetModel
-    global cur_style
-    # global causal_dit
-    # global controlnet
     weight_dtype = torch.float16
 
     if style == 'line':
@@ -388,7 +381,6 @@ 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)
 
@@ -427,8 +419,6 @@ def colorize_image(extracted_line, reference_images, resolution, seed, num_infer
     if extracted_line is None:
         gr.Info("Please preprocess the image first")
         raise ValueError("Please preprocess the image first")
-    global pipeline
-    global MultiResNetModel
     reference_images = process_multi_images(reference_images)
     fix_random_seeds(seed)