Upload FastAPI embedding app

Dockerfile

@@ -0,0 +1,17 @@
+FROM python:3.10-slim
+
+# Tạo thư mục làm việc
+WORKDIR /app
+
+# Copy tất cả file vào container
+COPY . /app
+RUN mkdir -p /app/cache
+
+# Cài thư viện Python
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Mở port cho API
+EXPOSE 7860
+
+# Chạy server FastAPI
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -0,0 +1,25 @@
+from fastapi import FastAPI
+from pydantic import BaseModel
+from sentence_transformers import SentenceTransformer
+import os
+os.environ['TRANSFORMERS_CACHE'] = '/app/cache'
+
+app = FastAPI(
+    title="Text Embedding API",
+    description="Dùng mô hình Vietnamese_Embedding từ AITeamVN",
+    version="1.0"
+)
+
+# Load mô hình
+model = SentenceTransformer("AITeamVN/Vietnamese_Embedding")
+model.max_seq_length = 2048
+
+# Định nghĩa schema input
+class TextInput(BaseModel):
+    texts: list[str]
+
+@app.post("/embed")
+def embed_text(data: TextInput):
+    inputs = [t.strip() for t in data.texts]
+    embs = model.encode(inputs, convert_to_numpy=True)
+    return {"embeddings": [e.tolist() for e in embs]}

requirements.txt

@@ -0,0 +1,4 @@
+fastapi
+uvicorn
+sentence-transformers
+pydantic