Spaces:
Runtime error
Runtime error
| from sentence_transformers import SentenceTransformer | |
| import numpy as np | |
| def generate_embeddings(texts): | |
| model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2') | |
| embeddings = model.encode(texts, convert_to_tensor=True) | |
| return embeddings | |
| def calculate_similarity(embedding1, embedding2): | |
| cosine_similarity = np.dot(embedding1, embedding2) / (np.linalg.norm(embedding1) * np.linalg.norm(embedding2)) | |
| return cosine_similarity.item() |