reverted NER model

README.md

@@ -6,7 +6,7 @@ colorTo: purple
 sdk: streamlit
 sdk_version: 1.2.0
 app_file: app.py
-models: ["Jean-Baptiste/roberta-large-ner-english"]
+models: ["dslim/bert-base-NER"]
 pinned: false
 license: mit
 ---

app.py

@@ -27,8 +27,8 @@ COLOR = {
 @st.cache(allow_output_mutation=True, show_spinner=True)
 def load_models():
     # NER
-    tokenizer = AutoTokenizer.from_pretrained("Jean-Baptiste/roberta-large-ner-english")
-    bert_ner = AutoModelForTokenClassification.from_pretrained("Jean-Baptiste/roberta-large-ner-english")
+    tokenizer = AutoTokenizer.from_pretrained("dslim/bert-base-NER")
+    bert_ner = AutoModelForTokenClassification.from_pretrained("dslim/bert-base-NER")
     tagger = pipeline("token-classification", model=bert_ner, tokenizer=tokenizer,
                       device=-1, aggregation_strategy="average")
     # NED