Update app.py

app.py

@@ -1,18 +1,19 @@
 import gradio as gr
 from fastapi import FastAPI, Request
 from fastapi.responses import JSONResponse
-from transformers import pipeline, AutoTokenizer
+from transformers import pipeline, AutoTokenizer, AutoModelForTokenClassification
 
-model = pipeline("token-classification",
-    model="ckiplab/bert-base-chinese-ner",
+# 正確載入 safetensors 模型
+model = pipeline(
+    "ner",
+    model=AutoModelForTokenClassification.from_pretrained("ckiplab/bert-base-chinese-ner", use_safetensors=True),
     tokenizer=AutoTokenizer.from_pretrained("ckiplab/bert-base-chinese-ner"),
-    aggregation_strategy="simple",
-    use_safetensors=True
+    aggregation_strategy="simple"
 )
 
 def analyze(sentence: str):
     result = model(sentence)
-    return  
+    return " ".join([f"{r['word']}/{r['entity_group']}" for r in result])
 
 demo = gr.Interface(fn=analyze, inputs="text", outputs="text", title="實體辨識")