AB

app.py

@@ -1,7 +1,12 @@
-from fastapi import FastAPI
+from transformers import AutoTokenizer, AutoModelForTokenClassification
+from transformers import pipeline
 
-app = FastAPI()
+tokenizer = AutoTokenizer.from_pretrained("Babelscape/wikineural-multilingual-ner")
+model = AutoModelForTokenClassification.from_pretrained("Babelscape/wikineural-multilingual-ner")
 
-@app.get("/")
-def greet_json():
-    return {"Hello": "World!"}
+nlp = pipeline("ner", model=model, tokenizer=tokenizer, grouped_entities=True)
+example = "张伟和李娜去北京旅行。"
+
+ner_results = nlp(example)
+print(example)
+print(ner_results)