Update app.py
Browse files
app.py
CHANGED
|
@@ -38,7 +38,7 @@ max_attempts = 10
|
|
| 38 |
def load_ner_model():
|
| 39 |
"""Loads the pre-trained NER model and caches it."""
|
| 40 |
|
| 41 |
-
return pipeline("token-classification", model="
|
| 42 |
|
| 43 |
|
| 44 |
# --- UI Elements ---
|
|
|
|
| 38 |
def load_ner_model():
|
| 39 |
"""Loads the pre-trained NER model and caches it."""
|
| 40 |
|
| 41 |
+
return pipeline("token-classification", model="ckiplab/albert-tiny-chinese-ner", aggregation_strategy="average")
|
| 42 |
|
| 43 |
|
| 44 |
# --- UI Elements ---
|