app.py

from fastapi import FastAPI
from pydantic import BaseModel
from sentence_transformers import SentenceTransformer

# FastAPI 애플리케이션 생성
app = FastAPI()

# 모델 로드
model = SentenceTransformer("sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2")

# 요청 형식 정의
class TextRequest(BaseModel):
    text: str

# API 엔드포인트 설정
@app.post("/embed")
async def embed_text(request: TextRequest):
    embedding = model.encode(request.text).tolist()
    return {"embedding": embedding}

# 기본 메시지
@app.get("/")
async def root():
    return {"message": "Text Embedding API is running"}