Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Update similarity.py
Browse files- similarity.py +1 -0
similarity.py
CHANGED
@@ -108,6 +108,7 @@ def check_similarity(images: List[RequestModel]):
|
|
108 |
logging.error(f"Error loading image for resource id {images[i].originId} : {e}")
|
109 |
similarity_score = 0
|
110 |
similarity_orb_score = 0
|
|
|
111 |
|
112 |
response = ResponseModel(originId=images[i].originId, source=images[i].source, sequence=images[i].sequence,
|
113 |
assetCode=images[i].assetCode, similarity=similarity_score, similarityOrb=similarity_orb_score, similarityMobileNet=similarity_mobilenet_score)
|
|
|
108 |
logging.error(f"Error loading image for resource id {images[i].originId} : {e}")
|
109 |
similarity_score = 0
|
110 |
similarity_orb_score = 0
|
111 |
+
similarity_mobilenet_score = 0
|
112 |
|
113 |
response = ResponseModel(originId=images[i].originId, source=images[i].source, sequence=images[i].sequence,
|
114 |
assetCode=images[i].assetCode, similarity=similarity_score, similarityOrb=similarity_orb_score, similarityMobileNet=similarity_mobilenet_score)
|