diff --git a/dlib/dlib_models_here.txt b/dlib/dlib_models_here.txt
deleted file mode 100644
index e69de29..0000000
diff --git a/faceanalysis.py b/faceanalysis.py
index 392a97f..2f8a350 100644
--- a/faceanalysis.py
+++ b/faceanalysis.py
@@ -188,7 +188,7 @@ class FaceEmbedDistance:
 
         ref = []
         for i in reference:
-            ref_emb = self.get_descriptor(np.array(T.ToPILImage()(i.permute(2, 0, 1)).convert('RGB')))
+            ref_emb = self.get_descriptor(np.array(T.ToPILImage()(i.permute(2, 0, 1).cpu()).convert('RGB')))
             if ref_emb is not None:
                 ref.append(torch.from_numpy(ref_emb))
         
@@ -203,7 +203,7 @@ class FaceEmbedDistance:
         out_cos = []
         
         for i in image:
-            img = np.array(T.ToPILImage()(i.permute(2, 0, 1)).convert('RGB'))
+            img = np.array(T.ToPILImage()(i.permute(2, 0, 1).cpu()).convert('RGB'))
 
             img = self.get_descriptor(img)