Spaces:

ResearchEngineering
/

news_sentiment_analyzer

Runtime error

ResearchEngineering commited on Aug 1

Commit

ddebd3a

verified ·

1 Parent(s): 4b67bac

Update src/streamlit_app.py

Files changed (1) hide show

src/streamlit_app.py CHANGED Viewed

@@ -574,10 +574,10 @@ def load_financial_models():
         # Load Financial NER model and tokenizer explicitly
         ner_tokenizer = AutoTokenizer.from_pretrained(
-            "elastic/distilbert-base-cased-finetuned-conll03-english", cache_dir=cache_dir
         )
         ner_model = AutoModelForTokenClassification.from_pretrained(
-            "elastic/distilbert-base-cased-finetuned-conll03-english", cache_dir=cache_dir
         )
         # Then create pipeline using objects

         # Load Financial NER model and tokenizer explicitly
         ner_tokenizer = AutoTokenizer.from_pretrained(
+            "Jean-Baptiste/roberta-large-ner-english", cache_dir=cache_dir
         )
         ner_model = AutoModelForTokenClassification.from_pretrained(
+            "Jean-Baptiste/roberta-large-ner-english", cache_dir=cache_dir
         )
         # Then create pipeline using objects