theArijitDas
commited on
Commit
•
c0d1cd6
1
Parent(s):
40f14c1
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
|