Spaces:
Runtime error
Runtime error
| from sentence_transformers import SentenceTransformer | |
| def load_model(device="cpu"): | |
| return SentenceTransformer("all-MiniLM-L6-v2").to(device) | |
| def compute_query_embedding(model, query): | |
| return model.encode(query, convert_to_tensor=True).cpu().numpy() | |