Fixes

Makefile

@@ -0,0 +1,2 @@
+test:
+	python -m pytest -vv

app.py

@@ -1,6 +1,7 @@
 import os
 from typing import Union
 
+import numpy as np
 from fastapi import Depends, FastAPI, File, HTTPException, Response, UploadFile, status
 from fastapi.security import APIKeyQuery
 
@@ -42,7 +43,7 @@ async def calculate_embedding(
     try:
         image_content = await image.read()
         pred = model.predict(model.preprocess(image_content))
-        return EmbeddingResponse(embedding=pred.tolist())
+        return EmbeddingResponse(embedding=pred[0].tolist())
     except Exception as e:
         return Response(
             content=str(e), status_code=status.HTTP_500_INTERNAL_SERVER_ERROR
@@ -62,9 +63,12 @@ async def calculate_similarity_score(
     try:
         image1_content = await image1.read()
         image2_content = await image2.read()
-        pred1 = model.predict(model.preprocess(image1_content))
-        pred2 = model.predict(model.preprocess(image2_content))
-        return SimilarityResponse(score=float(model.distance(pred1, pred2)))
+        pred = model.predict(
+            np.vstack(
+                [model.preprocess(image1_content), model.preprocess(image2_content)]
+            )
+        )
+        return SimilarityResponse(score=float(model.distance(pred[0], pred[1])))
     except Exception as e:
         return Response(
             content=str(e), status_code=status.HTTP_500_INTERNAL_SERVER_ERROR

test_service.py

@@ -45,13 +45,13 @@ def test_embed(client: TestClient, image_path):
         pytest.param(
             "testdata/face_pic.jpeg",
             "testdata/face_pic3.jpeg",
-            0.00026,
+            0.0588636,
             id="similar expression",
         ),
         pytest.param(
             "testdata/face_pic.jpeg",
             "testdata/face_pic2.jpeg",
-            2,
+            1.413582,
             id="different expression",
         ),
     ],