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)
|