|
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()} |
|
|
