Update similarity.py

similarity.py

@@ -163,14 +163,14 @@ def load_image_url(source, assetCode, contentType=None, ffmpeg_path='ffmpeg', fr
 def check_similarity(images: List[RequestModel]):
     logging.info(f"Checking similarity for main source with resource id {images[0].originId}")
 
-    original_image = load_image_url(images[0].source, images[0].assetCode)
+    original_image = load_image_url(images[0].source, images[0].assetCode, images[0].contentType)
     original_image_shape = original_image.shape
 
     results = []
 
     for i in range(1, len(images)):
         try:
-            image = load_image_url(images[i].source, images[i].source)
+            image = load_image_url(images[i].source, images[i].assetCode, images[i].contentType)
             image = cv2.resize(image, original_image_shape[::-1])
         
             similarity_score = ssim_sim(original_image, image)