Update app.py

app.py

@@ -1,25 +1,12 @@
-import streamlit as st
 from transformers import AutoTokenizer, AutoModelForTokenClassification, pipeline
-from accelerate import init_empty_weights, load_checkpoint_and_dispatch
 
 model_name = "Nucha/Nucha_ITSkillNER_BERT"
-
-# โหลด tokenizer
 tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForTokenClassification.from_pretrained(model_name)
 
-# โหลด model แบบ meta และโหลด weights อย่างถูกต้อง
-with init_empty_weights():
-    model = AutoModelForTokenClassification.from_pretrained(model_name)
-
-model = load_checkpoint_and_dispatch(
-    model,
-    model_name,
-    device_map="auto"
-)
-
-# สร้าง NER pipeline
 ner_pipeline = pipeline("ner", model=model, tokenizer=tokenizer)
 
+
 # Mapping สีของ Entity
 ENTITY_COLORS = {
     "HSKILL": "#FFD700",  # ทักษะเชิงเทคนิค