Spaces:

ProzisTech
/

similarity

Running on CPU Upgrade

MarioPrzBasto commited on May 13

Commit

e5555a2

verified ·

1 Parent(s): 048dc8d

Update similarity.py (#3)

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)