Update product_update_validator.py

product_update_validator.py

@@ -2,14 +2,14 @@ from description_validator import Description_Validator
 from image_validator import Image_Validator
 
 class Update_Validator:
-    def __init__(self, text_model=None, image_model=None, threshold=0.7):
+    def __init__(self, text_model=None, image_model=None, threshold=0.75):
         self.description_validator  = Description_Validator(model_name=text_model)
         self.image_validator        = Image_Validator(model_name=image_model)
         self.threshold              = threshold
     
     def validate(self, text1, image1, text2, image2, threshold=None, return_score=False):
         description_similarity  = self.description_validator.similarity_score(text1, text2)
-        image_similarity        = self.image_validator.similarity_score(image_path_1, image_path_2)
+        image_similarity        = self.image_validator.similarity_score(image1, image2)
         similarity_score        = 0.75 * description_similarity + 0.25 * image_similarity
         
         if threshold is None: threshold=self.threshold