Spaces:
Runtime error
Runtime error
Update src/streamlit_app.py
Browse files- 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="
|
| 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:
|