Update app.py

app.py

@@ -4,6 +4,7 @@ import numpy as np
 from skimage.metrics import structural_similarity as ssim
 from transformers import BertForSequenceClassification, BertTokenizer
 import torch
+from PIL import Image
 
 
 # Load pre-trained model and tokenizer
@@ -43,8 +44,8 @@ def calculate_color_similarity(img1, img2):
 
 
 def compare_trademarks(trademark1, trademark2):
-    img1 = cv2.imread(trademark1.name)
-    img2 = cv2.imread(trademark2.name)
+    img1 = np.array(trademark1)
+    img2 = np.array(trademark2)
 
     ssim_score = calculate_ssim(img1, img2)
 
@@ -67,8 +68,8 @@ def prevent_trademark_conflict(trademark1, trademark2):
 trademark_comparison_interface = gr.Interface(
     fn=prevent_trademark_conflict,
     inputs=[
-        gr.inputs.Image(type="file", label="Trademark Image 1"),
-        gr.inputs.Image(type="file", label="Trademark Image 2"),
+        gr.inputs.Image(type="pil", label="Trademark Image 1"),
+        gr.inputs.Image(type="pil", label="Trademark Image 2"),
     ],
     outputs="text",
     title="Trademark Comparison",