Chanjeans commited on
Commit
df082a5
ยท
verified ยท
1 Parent(s): 07e5493

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +24 -0
app.py ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from fastapi import FastAPI
2
+ from pydantic import BaseModel
3
+ from sentence_transformers import SentenceTransformer
4
+
5
+ # FastAPI ์• ํ”Œ๋ฆฌ์ผ€์ด์…˜ ์ƒ์„ฑ
6
+ app = FastAPI()
7
+
8
+ # ๋ชจ๋ธ ๋กœ๋“œ
9
+ model = SentenceTransformer("sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2")
10
+
11
+ # ์š”์ฒญ ํ˜•์‹ ์ •์˜
12
+ class TextRequest(BaseModel):
13
+ text: str
14
+
15
+ # API ์—”๋“œํฌ์ธํŠธ ์„ค์ •
16
+ @app.post("/embed")
17
+ async def embed_text(request: TextRequest):
18
+ embedding = model.encode(request.text).tolist()
19
+ return {"embedding": embedding}
20
+
21
+ # ๊ธฐ๋ณธ ๋ฉ”์‹œ์ง€
22
+ @app.get("/")
23
+ async def root():
24
+ return {"message": "Text Embedding API is running"}