Maria Tsilimos
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -108,7 +108,7 @@ if source_type:
|
|
108 |
@st.cache_resource
|
109 |
def load_ner_model():
|
110 |
|
111 |
-
return pipeline("token-classification", model="
|
112 |
|
113 |
model = load_ner_model()
|
114 |
experiment = None
|
|
|
108 |
@st.cache_resource
|
109 |
def load_ner_model():
|
110 |
|
111 |
+
return pipeline("token-classification", model="BSC-LT/roberta-base-bne-capitel-ner-plus", aggregation_strategy="max")
|
112 |
|
113 |
model = load_ner_model()
|
114 |
experiment = None
|