Create app.py

app.py

@@ -0,0 +1,20 @@
+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