Fix fecnet

model/fecnet.py

@@ -53,10 +53,10 @@ class FECNetModel:
         model_weights = torch.load(model_path, map_location=torch.device("cpu"))
         model.load_state_dict(model_weights)
         model.eval()
-        return model.double()
+        return model
 
     def predict(self, image: np.ndarray):
-        pred = self.model.forward(image)
+        pred = self.model(image)
         return pred
 
     def distance(a, b):
@@ -64,9 +64,10 @@ class FECNetModel:
 
     def embed_image(self, image) -> np.ndarray:
         image = cv2.imdecode(image, cv2.IMREAD_COLOR)
+        image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
         image = cv2.resize(image, (224, 224))
         image = np.transpose(image, (2, 0, 1))
         image = np.expand_dims(image, axis=0)
-        image = torch.from_numpy(image).double()
+        image = torch.from_numpy(image.astype(np.float32))
         pred = self.predict(image)
         return pred.detach().numpy()