Change score to cosine similarity

app.py

@@ -64,7 +64,9 @@ async def calculate_similarity_score(
                 [model.preprocess(image1_content), model.preprocess(image2_content)]
             )
         )
-        return SimilarityResponse(score=float(model.distance(pred[0], pred[1])))
+        return SimilarityResponse(
+            score=float(model.cosine_similarity(pred[0], pred[1]))
+        )
     except Exception as e:
         return Response(
             content=str(e), status_code=status.HTTP_500_INTERNAL_SERVER_ERROR

model/model.py

@@ -30,6 +30,10 @@ class Model:
     def distance(x1: np.ndarray, x2: np.ndarray):
         return np.linalg.norm(x1 - x2, ord=2)
 
+    @staticmethod
+    def cosine_similarity(x1: np.ndarray, x2: np.ndarray):
+        return np.dot(x1, x2) / (np.linalg.norm(x1) * np.linalg.norm(x2))
+
 
 class LandmarkExtractor:
     def __init__(self):