Upload 3 files

app.py

@@ -3,7 +3,12 @@ from fastapi import FastAPI, Request
 from fastapi.responses import JSONResponse
 from transformers import pipeline, AutoTokenizer
 
-model = pipeline("token-classification", model="ckiplab/bert-base-chinese-ner", tokenizer=AutoTokenizer.from_pretrained("ckiplab/bert-base-chinese-ner"), aggregation_strategy="simple")
+model = pipeline("token-classification",
+    model="ckiplab/bert-base-chinese-ner",
+    tokenizer=AutoTokenizer.from_pretrained("ckiplab/bert-base-chinese-ner"),
+    aggregation_strategy="simple",
+    use_safetensors=True
+)
 
 def analyze(sentence: str):
     result = model(sentence)