Update app.py

app.py

@@ -14,11 +14,12 @@ from huggingface_hub import hf_hub_download
 ### We use the ckpt of 79999_iter.pth: https://drive.google.com/open?id=154JgKpzCPW82qINcVieuPH3fZ2e0P812
 ### Thanks for the open source of face-parsing model.
 from models.BiSeNet.model import BiSeNet
-from multiprocessing import Process, Queue
+from multiprocessing import Process, Queue, Manager
 
-zero = torch.Tensor([0]).cuda()
-print(zero.device) # <-- 'cpu' 🤔
-device = zero.device # "cuda"
+# zero = torch.Tensor([0]).cuda()
+# print(zero.device) # <-- 'cpu' 🤔
+# device = zero.device # "cuda"
+device = "cuda"
 
 # Gets the absolute path of the current script
 script_directory = os.path.dirname(os.path.realpath(__file__))
@@ -38,7 +39,6 @@ pipe = ConsistentIDStableDiffusionPipeline.from_pretrained(
 ### 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))