modules/comparison.py

import torch.nn.functional as F

class Comparison:
    def __init__(self, model):
        self.model = model

    def encode(self, texts):
        return self.model.encode(texts, convert_to_tensor=True, normalize_embeddings=True)

    def compare(self, resume, jd):
        return F.cosine_similarity(jd, resume, dim=-1).tolist()