ResearchEngineering commited on
Commit
5ba72b0
·
verified ·
1 Parent(s): 6b1983c

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -570,7 +570,7 @@ def load_financial_models():
570
  finbert_model = AutoModelForSequenceClassification.from_pretrained("yiyanghkust/finbert-tone", cache_dir=cache_dir)
571
 
572
  # Financial NER for entity extraction
573
- ner_pipeline = pipeline("ner", model="dbmdz/bert-large-cased-finetuned-conll03-english", aggregation_strategy="simple", cache_dir=cache_dir)
574
 
575
  return finbert_tokenizer, finbert_model, ner_pipeline, None
576
  except Exception as e:
 
570
  finbert_model = AutoModelForSequenceClassification.from_pretrained("yiyanghkust/finbert-tone", cache_dir=cache_dir)
571
 
572
  # Financial NER for entity extraction
573
+ ner_pipeline = pipeline("ner", model="elastic/distilbert-base-cased-finetuned-conll03-english", aggregation_strategy="simple", cache_dir=cache_dir)
574
 
575
  return finbert_tokenizer, finbert_model, ner_pipeline, None
576
  except Exception as e: