sentence_similarity.py

# Make sure you have torch and sentence_transformers libraries installed
# 1 - Import Libraries & Install Files
import torch
from sentence_transformers import SentenceTransformer
from sentence_transformers import util 

#We will find similarity for these two sentences
sentence1 = "This is a sentence"
sentence2 = "This is also a sentence"

#Download model
model = SentenceTransformer('all-MiniLM-L6-v2')

#Find sentence embeddings for sentences using SBERT model
embedding_sen1 = model.encode(sentence1)
embedding_sen2 = model.encode(sentence2)

#Find cosine distance between the sentences
cos_sim = util.cos_sim(embedding_sen1, embedding_sen2)
print("Cosine-Similarity:", cos_sim.numpy()[0][0])
print("Percentage Similarity: ", cos_sim.numpy()[0][0]*100, "%")