| from fastapi import FastAPI |
| from pydantic import BaseModel |
| from typing import List |
| from sentence_transformers import SentenceTransformer |
| from sklearn.metrics.pairwise import cosine_similarity |
|
|
| app = FastAPI() |
|
|
| model = SentenceTransformer("paraphrase-MiniLM-L3-v2") |
|
|
| class SimilarityRequest(BaseModel): |
| ideal_profile: str |
| text_blocks: List[str] |
|
|
| @app.post("/similarity") |
| def compute_similarity(req: SimilarityRequest): |
| profile_vec = model.encode(req.ideal_profile) |
| text_vecs = model.encode(req.text_blocks) |
| similarities = cosine_similarity([profile_vec], text_vecs)[0].tolist() |
| return similarities |
|
|
