add model

app.py

@@ -1,7 +1,18 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, Request
+from pydantic import BaseModel
+from transformers import AutoTokenizer, AutoModelForTokenClassification, pipeline
 
 app = FastAPI()
 
-@app.get("/")
-def greet_json():
-    return {"Hello": "World!"}
+tokenizer = AutoTokenizer.from_pretrained("NlpHUST/vi-word-segmentation")
+model = AutoModelForTokenClassification.from_pretrained("NlpHUST/vi-word-segmentation")
+nlp = pipeline("token-classification", model=model, tokenizer=tokenizer)
+
+class InputText(BaseModel):
+    text: str
+
+@app.post("/segment")
+async def segment_text(payload: InputText):
+    text = payload.text
+    result = nlp(text)
+    return result

requirements.txt

@@ -1,2 +1,4 @@
 fastapi
-uvicorn[standard]
+uvicorn[standard]
+transformers
+torch