Update app.py
Browse files
app.py
CHANGED
@@ -7,8 +7,8 @@ import io
|
|
7 |
|
8 |
# Function to calculate SSIM between two images
|
9 |
def calculate_ssim(img1, img2):
|
10 |
-
img1_gray =
|
11 |
-
img2_gray =
|
12 |
return ssim(np.array(img1_gray), np.array(img2_gray))
|
13 |
|
14 |
# Function to compare two trademarks
|
@@ -25,8 +25,8 @@ def prevent_trademark_conflict(trademark1, trademark2):
|
|
25 |
iface = gr.Interface(
|
26 |
fn=prevent_trademark_conflict,
|
27 |
inputs=[
|
28 |
-
gr.inputs.Image(type="
|
29 |
-
gr.inputs.Image(type="
|
30 |
],
|
31 |
outputs="text",
|
32 |
title="Trademark Similarity",
|
|
|
7 |
|
8 |
# Function to calculate SSIM between two images
|
9 |
def calculate_ssim(img1, img2):
|
10 |
+
img1_gray = img1.convert("L")
|
11 |
+
img2_gray = img2.convert("L")
|
12 |
return ssim(np.array(img1_gray), np.array(img2_gray))
|
13 |
|
14 |
# Function to compare two trademarks
|
|
|
25 |
iface = gr.Interface(
|
26 |
fn=prevent_trademark_conflict,
|
27 |
inputs=[
|
28 |
+
gr.inputs.Image(type="pil", label="Trademark Image 1"),
|
29 |
+
gr.inputs.Image(type="pil", label="Trademark Image 2")
|
30 |
],
|
31 |
outputs="text",
|
32 |
title="Trademark Similarity",
|