Update app.py

app.py

@@ -14,7 +14,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForTokenClassification.from_pretrained(model_name)
 
 # สร้าง pipeline สำหรับ NER
-ner_pipeline = pipeline("ner", model=model, tokenizer=tokenizer, aggregation_strategy="simple")
+ner_pipeline = pipeline("ner", model=model, tokenizer=tokenizer)
 
 # UI ด้วย Streamlit
 st.title("NER Analysis with Nucha SkillNER BERT and Spark NLP Display")
@@ -22,7 +22,8 @@ st.title("NER Analysis with Nucha SkillNER BERT and Spark NLP Display")
 text = st.text_area("Enter text for NER analysis:")
 
 if st.button("Analyze"):
-    ner_results = ner_pipeline(text)
+    # สร้าง pipeline สำหรับ NER
+    ner_results = pipeline("ner", model=model, tokenizer=tokenizer)
 
     # สร้าง DataFrame สำหรับผลลัพธ์
     data = [{"word": entity['word'], "start": entity['start'], "end": entity['end'], "label": entity['entity']} for entity in ner_results]