Update app.py
Browse files
app.py
CHANGED
|
@@ -43,8 +43,8 @@ def calculate_color_similarity(img1, img2):
|
|
| 43 |
|
| 44 |
|
| 45 |
def compare_trademarks(trademark1, trademark2):
|
| 46 |
-
img1 = cv2.imread(trademark1)
|
| 47 |
-
img2 = cv2.imread(trademark2)
|
| 48 |
|
| 49 |
ssim_score = calculate_ssim(img1, img2)
|
| 50 |
|
|
@@ -67,8 +67,8 @@ def prevent_trademark_conflict(trademark1, trademark2):
|
|
| 67 |
trademark_comparison_interface = gr.Interface(
|
| 68 |
fn=prevent_trademark_conflict,
|
| 69 |
inputs=[
|
| 70 |
-
gr.inputs.Image(type="
|
| 71 |
-
gr.inputs.Image(type="
|
| 72 |
],
|
| 73 |
outputs="text",
|
| 74 |
title="Trademark Comparison",
|
|
|
|
| 43 |
|
| 44 |
|
| 45 |
def compare_trademarks(trademark1, trademark2):
|
| 46 |
+
img1 = cv2.imread(trademark1.name)
|
| 47 |
+
img2 = cv2.imread(trademark2.name)
|
| 48 |
|
| 49 |
ssim_score = calculate_ssim(img1, img2)
|
| 50 |
|
|
|
|
| 67 |
trademark_comparison_interface = gr.Interface(
|
| 68 |
fn=prevent_trademark_conflict,
|
| 69 |
inputs=[
|
| 70 |
+
gr.inputs.Image(type="file", label="Trademark Image 1"),
|
| 71 |
+
gr.inputs.Image(type="file", label="Trademark Image 2"),
|
| 72 |
],
|
| 73 |
outputs="text",
|
| 74 |
title="Trademark Comparison",
|