Fix shape

model/model.py

@@ -24,7 +24,7 @@ class Model:
         array_repr = np.asarray(bytearray(image), dtype=np.uint8)
         decoded_img = cv2.imdecode(array_repr, flags=cv2.IMREAD_COLOR)
         landmarks = self.landmark_extractor.extract_landmarks_flat(decoded_img)
-        return pd.DataFrame([landmarks]).to_numpy().reshape(478, -1)
+        return pd.DataFrame([landmarks]).to_numpy().reshape(1, 478, -1)
 
     @staticmethod
     def distance(x1: np.ndarray, x2: np.ndarray):