Spaces:

ProzisTech
/

similarity

Running on CPU Upgrade

MarioPrzBasto commited on 4 days ago

Commit

f49303f

verified ·

1 Parent(s): 9a8a292

Update similarity.py

Files changed (1) hide show

similarity.py CHANGED Viewed

@@ -183,7 +183,7 @@ def check_similarity(images: List[RequestModel]):
                 similarity_orb_score = 0
                 similarity_mobilenet_score = 0
-            response = ResponseModel(originId=images[i].originId, source=None, sequence=images[i].sequence,
                                      assetCode=images[i].assetCode, similarity=similarity_score, similarityOrb=similarity_orb_score, similarityMobileNet=similarity_mobilenet_score)
             results.append(response)
@@ -195,7 +195,7 @@ def check_similarity(images: List[RequestModel]):
             similarity_orb_score = 0
             similarity_mobilenet_score = 0
-            response = ResponseModel(originId=images[i].originId, source=None, sequence=images[i].sequence,
                                      assetCode=images[i].assetCode, similarity=similarity_score, similarityOrb=similarity_orb_score, similarityMobileNet=similarity_mobilenet_score)
             results.append(response)

                 similarity_orb_score = 0
                 similarity_mobilenet_score = 0
+            response = ResponseModel(originId=images[i].originId, sequence=images[i].sequence,
                                      assetCode=images[i].assetCode, similarity=similarity_score, similarityOrb=similarity_orb_score, similarityMobileNet=similarity_mobilenet_score)
             results.append(response)
             similarity_orb_score = 0
             similarity_mobilenet_score = 0
+            response = ResponseModel(originId=images[i].originId, sequence=images[i].sequence,
                                      assetCode=images[i].assetCode, similarity=similarity_score, similarityOrb=similarity_orb_score, similarityMobileNet=similarity_mobilenet_score)
             results.append(response)