Add cosine sim

app.py

@@ -31,7 +31,9 @@ def compare(path1, path2):
     #     return OUTPUT_OK.format(similarity_score * 100)
     # else:
     #     return OUTPUT_FAIL.format(similarity_score * 100)
-    return "{:.4f}".format(similarity_score.item())
+    cosine_sim = torch.nn.CosineSimilarity(dim=-1)
+    cosine_similiarity = cosine_sim(embs1, embs2)
+    return "Calulated Score: {:.2f}\nCosineSimilarity: {:.4f}".format(similarity_score.item() * 100, cosine_similiarity)
 
 
 inputs = [