File size: 763 Bytes
ca8bacf ed7f561 ca8bacf ed7f561 ca8bacf f3e3788 ed7f561 f3e3788 ed7f561 f3e3788 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
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"""
<html>
<body>
<h1>Hello, World!</h1>
<a href='/generate-embeddings/sentence-transformers-all-MiniLM-L6-v2/'>Generate Embeddings</a>
</body>
</html>
"""
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
|