LinkLinkWu commited on
Commit
f38602e
·
verified ·
1 Parent(s): 35b50a1

Update func.py

Browse files
Files changed (1) hide show
  1. func.py +1 -1
func.py CHANGED
@@ -9,7 +9,7 @@ _ner_pipeline = None
9
  def get_sentiment_pipeline():
10
  global _sentiment_pipeline
11
  if _sentiment_pipeline is None:
12
- model_id = "LinkLinkWu/ISOM5240TRAIN2"
13
  tokenizer = AutoTokenizer.from_pretrained(model_id)
14
  model = AutoModelForSequenceClassification.from_pretrained(model_id)
15
  _sentiment_pipeline = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
 
9
  def get_sentiment_pipeline():
10
  global _sentiment_pipeline
11
  if _sentiment_pipeline is None:
12
+ model_id = "LinkLinkWu/ISOM5240HKUSTBASE"
13
  tokenizer = AutoTokenizer.from_pretrained(model_id)
14
  model = AutoModelForSequenceClassification.from_pretrained(model_id)
15
  _sentiment_pipeline = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)