Update app.py

app.py

@@ -36,6 +36,14 @@ pipe = ConsistentIDStableDiffusionPipeline.from_pretrained(
 ).to(device)
 
 ### Load other pretrained models
+@spaces.GPU
+def load_model(queue, bise_net_cp_path):
+    device = torch.device('cuda')
+    bise_net = BiSeNet(n_classes = 19)
+    bise_net.to(device)
+    bise_net.load_state_dict(torch.load(bise_net_cp_path))
+    bise_net.eval()
+    queue.put(bise_net)
 ## BiSenet
 bise_net_cp_path = hf_hub_download(repo_id="JackAILab/ConsistentID", filename="face_parsing.pth", repo_type="model")
 # Create a queue to share data between processes
@@ -58,15 +66,6 @@ pipe.load_ConsistentID_model(
 pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
 
 
-@spaces.GPU
-def load_model(queue, bise_net_cp_path):
-    device = torch.device('cuda')
-    bise_net = BiSeNet(n_classes = 19)
-    bise_net.to(device)
-    bise_net.load_state_dict(torch.load(bise_net_cp_path))
-    bise_net.eval()
-    queue.put(bise_net)
-
 @spaces.GPU
 def process(inputImage,prompt,negative_prompt):