edusocial's picture
Update README.md
1e3674a verified
|
Raw
History Blame Contribute Delete
1.14 kB
---
title: 🧠 intfloat/e5-small Embedding API Server
emoji: 🧩
colorFrom: blue
colorTo: purple
sdk: docker
app_file: app.py
pinned: false
---
# 🧠 Local Embedding API Server (Multilingual)
A lightweight embedding generation server using FastAPI + SentenceTransformer, supporting Chinese, English, Malay and more β€” powered by `BAAI/bge-m3`.
## πŸš€ Features
- `/embed`: generate embedding for a single string
- `/embed/batch`: generate embedding for multiple strings
- Support for GPU acceleration (CUDA) via Docker
- Docker-ready deployment
## πŸ”§ Requirements
- Docker (with NVIDIA container runtime)
- Optional: Python 3.10+ for local dev
## βš™οΈ Run with Docker (GPU)
```bash
docker build -t embed-server .
docker run --gpus all --env-file .env -p 8000:8000 embed-server
```
## πŸ“¦ API Usage
### `POST /embed`
```json
{
"input": "Saya suka makan nasi lemak"
}
```
### `POST /embed/batch`
```json
{
"inputs": ["I like cats", "ζˆ‘ε–œζ¬’ηŒ«", "Saya suka kucing"]
}
```
## πŸ“€ Response
```json
{
"embedding": [...],
"dim": 1024,
"model": "altaidevorg/BGE-M3-Distill-8L"
}
```
## ✍️ License
MIT