Update app.py

app.py

@@ -55,6 +55,10 @@ def compute_similarity(input1, input2, text1, text2, type1, type2):
         else:
             embedding2 = model.get_text_features(input_ids=input2_tensor)
     
+    # Normalize embeddings
+    embedding1 = embedding1 / embedding1.norm(dim=-1, keepdim=True)
+    embedding2 = embedding2 / embedding2.norm(dim=-1, keepdim=True)
+    
     # Compute cosine similarity
     similarity = torch.nn.functional.cosine_similarity(embedding1, embedding2).item()
     return f"Similarity Score: {similarity:.4f}"