Spaces:
Runtime error
Runtime error
Atharva
commited on
Commit
•
fe0139a
1
Parent(s):
f441aa5
reverted NER model
Browse files
README.md
CHANGED
@@ -6,7 +6,7 @@ colorTo: purple
|
|
6 |
sdk: streamlit
|
7 |
sdk_version: 1.2.0
|
8 |
app_file: app.py
|
9 |
-
models: ["
|
10 |
pinned: false
|
11 |
license: mit
|
12 |
---
|
|
|
6 |
sdk: streamlit
|
7 |
sdk_version: 1.2.0
|
8 |
app_file: app.py
|
9 |
+
models: ["dslim/bert-base-NER"]
|
10 |
pinned: false
|
11 |
license: mit
|
12 |
---
|
app.py
CHANGED
@@ -27,8 +27,8 @@ COLOR = {
|
|
27 |
@st.cache(allow_output_mutation=True, show_spinner=True)
|
28 |
def load_models():
|
29 |
# NER
|
30 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
31 |
-
bert_ner = AutoModelForTokenClassification.from_pretrained("
|
32 |
tagger = pipeline("token-classification", model=bert_ner, tokenizer=tokenizer,
|
33 |
device=-1, aggregation_strategy="average")
|
34 |
# NED
|
|
|
27 |
@st.cache(allow_output_mutation=True, show_spinner=True)
|
28 |
def load_models():
|
29 |
# NER
|
30 |
+
tokenizer = AutoTokenizer.from_pretrained("dslim/bert-base-NER")
|
31 |
+
bert_ner = AutoModelForTokenClassification.from_pretrained("dslim/bert-base-NER")
|
32 |
tagger = pipeline("token-classification", model=bert_ner, tokenizer=tokenizer,
|
33 |
device=-1, aggregation_strategy="average")
|
34 |
# NED
|