Spaces:
Build error
Build error
Update product_update_validator.py
Browse files
product_update_validator.py
CHANGED
|
@@ -7,7 +7,7 @@ class Update_Validator:
|
|
| 7 |
self.image_validator = Image_Validator(model_name=image_model)
|
| 8 |
self.threshold = threshold
|
| 9 |
|
| 10 |
-
def validate(self, text1, image1, text2,
|
| 11 |
description_similarity = self.description_validator.similarity_score(text1, text2)
|
| 12 |
image_similarity = self.image_validator.similarity_score(image_path_1, image_path_2)
|
| 13 |
similarity_score = 0.75 * description_similarity + 0.25 * image_similarity
|
|
|
|
| 7 |
self.image_validator = Image_Validator(model_name=image_model)
|
| 8 |
self.threshold = threshold
|
| 9 |
|
| 10 |
+
def validate(self, text1, image1, text2, image2, threshold=None, return_score=False):
|
| 11 |
description_similarity = self.description_validator.similarity_score(text1, text2)
|
| 12 |
image_similarity = self.image_validator.similarity_score(image_path_1, image_path_2)
|
| 13 |
similarity_score = 0.75 * description_similarity + 0.25 * image_similarity
|