omar87 commited on
Commit
48e5876
·
1 Parent(s): 42b03f9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -14,11 +14,14 @@ def calculate_similarity(img1, img2):
14
  # Function to compute similarity scores for all images
15
  def compute_similarity(target_image, image_list):
16
  scores = []
 
17
  for image in image_list:
18
- similarity_score = calculate_similarity(target_image, image)
 
19
  scores.append(similarity_score)
20
  return scores
21
 
 
22
  # Function to handle the Gradio interface
23
  def image_similarity(target_image, image_list):
24
  target_image = target_image.astype(np.uint8)
 
14
  # Function to compute similarity scores for all images
15
  def compute_similarity(target_image, image_list):
16
  scores = []
17
+ target_image_resized = cv2.resize(target_image, (target_image.shape[1], target_image.shape[0]))
18
  for image in image_list:
19
+ image_resized = cv2.resize(image, (target_image.shape[1], target_image.shape[0]))
20
+ similarity_score = calculate_similarity(target_image_resized, image_resized)
21
  scores.append(similarity_score)
22
  return scores
23
 
24
+
25
  # Function to handle the Gradio interface
26
  def image_similarity(target_image, image_list):
27
  target_image = target_image.astype(np.uint8)