lmcuong199 commited on
Commit
105ae96
·
verified ·
1 Parent(s): 8240af2

Update nlp_module.py

Browse files
Files changed (1) hide show
  1. nlp_module.py +1 -1
nlp_module.py CHANGED
@@ -12,7 +12,7 @@ def get_classifier():
12
  if _classifier is None:
13
  _classifier = pipeline(
14
  "text-classification",
15
- model="climatebert/distilroberta-base-climate-sentiment",
16
  top_k=None
17
  )
18
  return _classifier
 
12
  if _classifier is None:
13
  _classifier = pipeline(
14
  "text-classification",
15
+ model="distilbert-base-uncased-finetuned-sst-2-english",
16
  top_k=None
17
  )
18
  return _classifier