Update app.py

app.py

@@ -28,8 +28,8 @@ def calculate_text_similarity(text1, text2):
         outputs_text1 = model(**encoded_text1)
         outputs_text2 = model(**encoded_text2)
 
-    embeddings_text1 = outputs_text1.last_hidden_state.squeeze(0)
-    embeddings_text2 = outputs_text2.last_hidden_state.squeeze(0)
+    embeddings_text1 = outputs_text1.pooler_output.squeeze(0)
+    embeddings_text2 = outputs_text2.pooler_output.squeeze(0)
 
     text_similarity = ssim(embeddings_text1.numpy(), embeddings_text2.numpy())
 
@@ -72,6 +72,7 @@ def prevent_trademark_conflict(trademark1, trademark2):
     return similarity_scores
 
 
+# Interface
 trademark_comparison_interface = gr.Interface(
     fn=prevent_trademark_conflict,
     inputs=[
@@ -80,9 +81,8 @@ trademark_comparison_interface = gr.Interface(
     ],
     outputs="text",
     title="Trademark Comparison",
-    description="Compare two trademarks based on SSIM and text similarity.",
+    description="Compare two trademarks based on SSIM, text similarity, and color similarity.",
 )
 
 # Launch the interface
 trademark_comparison_interface.launch()
-