Update main.py

main.py

@@ -5,6 +5,7 @@ from PIL import Image
 import io
 import base64
 from sentence_transformers import SentenceTransformer, util
+import json
 
 class Item(BaseModel):
     name: str
@@ -72,7 +73,11 @@ def predict(item: Item):
     for score, image_id1, image_id2 in near_duplicates[0:NUM_SIMILAR_IMAGES]:
         print("\nScore: {:.3f}%".format(score * 100))
 
-    return "Score: {:.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 "Score: {:.3f}%".format(score * 100)
 
 #def update_item(item: Item):
 #   return item