BookRecommend / app.py
AmiKim's picture
Create app.py
044074a verified
raw
history blame contribute delete
384 Bytes
from fastapi import FastAPI
from pydantic import BaseModel
from kiwipiepy import Kiwi
app = FastAPI()
kiwi = Kiwi()
class TextInput(BaseModel):
text: str
@app.post("/tokenize")
def tokenize(input: TextInput):
tokens = kiwi.tokenize(input.text)
return {
"tokens": [t.form for t in tokens],
"tagged": [{"form": t.form, "tag": t.tag} for t in tokens]
}