from fastapi import FastAPI
from sentence_transformers import SentenceTransformer
app = FastAPI()
all_MiniLM_L6_V2_model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
@app.get("/")
def greet_json():
return f"""
Hello, World!
Generate Embeddings
"""
from typing import List
@app.post("/generate-embeddings/sentence-transformers-all-MiniLM-L6-v2/")
def generate_embeddings_all_MiniLM_L6_V2_model(sentences: List[str]):
embeddings = all_MiniLM_L6_V2_model.encode(sentences)
return {"embeddings": embeddings.tolist()} # Return embeddings as a JSON-compatible list