Similarity.py

from sentence_transformers import SentenceTransformer

class Similarity:
    def __init__(self):
        self.model = None

    def load_model(self):
        if self.model is None:
            print("Loading SentenceTransformer model...")
            self.model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
            print("Model loaded.")

    def embed_text(self, text):
        self.load_model()
        return self.model.encode(text, convert_to_tensor=True)