Update main.py

main.py

@@ -62,7 +62,7 @@ def mobilenet_similarity(img1, img2):
         sim = cosine_similarity(feat1, feat2)[0][0]  # Valor entre -1 e 1
         sim_score = (sim + 1) * 50  # Escalar para 0-100
         print(f"MobileNet similarity score is {sim_score}")
-        return sim_score
+        return float(sim_score)
     except Exception as e:
         print("Erro ao calcular similaridade com MobileNet")
         return 0
@@ -106,7 +106,7 @@ async def save(image_data: RequestModel):
     data_to_save = image_data.dict()
     if similarity_orb is not None:
         data_to_save["similarityOrb"] = similarity_orb
-        data_to_save["similarityMobilenet"] = float(similarity_mobilenet)
+        data_to_save["similarityMobilenet"] = similarity_mobilenet
 
     with open(filename, "w") as f:
         json.dump(data_to_save, f, indent=4)