omar87 commited on
Commit
f9911e4
·
1 Parent(s): 231b140

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -38,7 +38,12 @@ def compare_colors(trademark1, trademark2):
38
 
39
  # Function to compare trademarks based on multiple aspects
40
  def compare_trademarks(trademark1, trademark2):
41
- ssim_score = calculate_ssim(np.array(trademark1), np.array(trademark2))
 
 
 
 
 
42
  text_similarity = compare_text(trademark1, trademark2)
43
  color_similarity = compare_colors(trademark1, trademark2)
44
 
@@ -64,6 +69,7 @@ iface = gr.Interface(
64
  ],
65
  outputs="text",
66
  title="Trademark Conflict Prevention",
 
67
  description="Upload two trademark images to prevent conflict."
68
  )
69
 
 
38
 
39
  # Function to compare trademarks based on multiple aspects
40
  def compare_trademarks(trademark1, trademark2):
41
+ # Resize images to a consistent size
42
+ size = (300, 300)
43
+ trademark1_resized = trademark1.resize(size)
44
+ trademark2_resized = trademark2.resize(size)
45
+
46
+ ssim_score = calculate_ssim(np.array(trademark1_resized), np.array(trademark2_resized))
47
  text_similarity = compare_text(trademark1, trademark2)
48
  color_similarity = compare_colors(trademark1, trademark2)
49
 
 
69
  ],
70
  outputs="text",
71
  title="Trademark Conflict Prevention",
72
+
73
  description="Upload two trademark images to prevent conflict."
74
  )
75