Update utils.py

utils.py

@@ -17,11 +17,11 @@ THRESHOLD = 0.8
 mtcnn = MTCNN(
     image_size=160, margin=0, min_face_size=20,
     thresholds=[0.6, 0.7, 0.7], factor=0.709, post_process=True,
-    device=device,
+    device='cpu',
     keep_all=True
 )
 
-resnet = InceptionResnetV1(pretrained='vggface2').eval().to(device)
+resnet = InceptionResnetV1(pretrained='vggface2').eval().to('cpu')
 
 def extract_face(box, img, margin=20):
     img_size = (len(img), len(img[0]))
@@ -52,9 +52,9 @@ def trans(img):
 
 def find_and_compare(model, face):
     face = trans(face)
-    face_embedded = model(face.unsqueeze(dim=0).to(device))
+    face_embedded = model(face.unsqueeze(dim=0).to('cpu'))
     for usr in os.listdir(SINGLE_FACE_PATH):
-        usr_embedding = torch.load(SINGLE_FACE_PATH + "/" + usr + "/" + usr + ".pt").to(device)
+        usr_embedding = torch.load(SINGLE_FACE_PATH + "/" + usr + "/" + usr + ".pt").to('cpu')
         if (face_embedded - usr_embedding).norm().item() < THRESHOLD:
             print("Found " + usr + " in the image!")
             return usr