LinkLinkWu commited on
Commit
bdaffbb
·
verified ·
1 Parent(s): 628c80f

Update func.py

Browse files
Files changed (1) hide show
  1. func.py +1 -1
func.py CHANGED
@@ -4,7 +4,7 @@ import requests
4
 
5
  # ----------- Eager Initialization of Pipelines -----------
6
  # Sentiment pipeline
7
- model_id = "LinkLinkWu/ISOM5240HKUSTBASE"
8
  sentiment_tokenizer = AutoTokenizer.from_pretrained(model_id)
9
  sentiment_model = AutoModelForSequenceClassification.from_pretrained(model_id)
10
  sentiment_pipeline = pipeline(
 
4
 
5
  # ----------- Eager Initialization of Pipelines -----------
6
  # Sentiment pipeline
7
+ model_id = "ahmedrachid/FinancialBERT-Sentiment-Analysis"
8
  sentiment_tokenizer = AutoTokenizer.from_pretrained(model_id)
9
  sentiment_model = AutoModelForSequenceClassification.from_pretrained(model_id)
10
  sentiment_pipeline = pipeline(