Spaces:

ProzisTech
/

similarity

Running on CPU Upgrade

MarioPrzBasto commited on May 13

Commit

4de77ca

verified ·

1 Parent(s): b9177b9

Update similarity.py

Files changed (1) hide show

similarity.py CHANGED Viewed

@@ -44,7 +44,7 @@ def check_similarity(images: List[RequestModel]):
             logging.error(f"Error loading image for resource id {images[i].originId} : {e}")
             similarity_score = 0
-        response = ResponseModel(originId=images[i].originId, sequence=images[i].sequence,
                                  assetCode=images[i].assetCode, similarity=similarity_score)
         results.append(response)

             logging.error(f"Error loading image for resource id {images[i].originId} : {e}")
             similarity_score = 0
+        response = ResponseModel(originId=images[i].originId, source=images[i].source, sequence=images[i].sequence,
                                  assetCode=images[i].assetCode, similarity=similarity_score)
         results.append(response)