Spaces:
Running
Running
File size: 568 Bytes
1440496 8c3ae63 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
import os
os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
os.makedirs("/tmp/hf_cache", exist_ok=True)
from fastapi import FastAPI, Request
from sentence_transformers import SentenceTransformer
import uvicorn
app = FastAPI()
model = SentenceTransformer('intfloat/multilingual-e5-base')
@app.post("/embed")
async def embed(request: Request):
body = await request.json()
text = body.get("text", "")
embedding = model.encode(text).tolist()
return {"embedding": embedding}
if __name__ == "__main__":
uvicorn.run(app, host="0.0.0.0", port=8000)
|