Update main.py

main.py

@@ -5,13 +5,15 @@ from PIL import Image
 import io
 import base64
 from sentence_transformers import SentenceTransformer, util
-import json
 
 class Item(BaseModel):
     name: str
     image1: bytes
     image2: bytes
 
+class ScoreResponse(BaseModel):
+    score: float
+
 app = FastAPI()
 
 @app.get("/")
@@ -74,8 +76,7 @@ def predict(item: Item):
         print("\nScore: {:.3f}%".format(score * 100))
 
     formatted_score = round(score * 100, 3)  # Multiplies by 100 and rounds to three decimal places
-    score_json = json.dumps({"score": formatted_score})
-    return score_json
+    return ScoreResponse(score=formatted_score)
     
     #return "Score: {:.3f}%".format(score * 100)