sm2899 commited on
Commit
eead5f7
1 Parent(s): 12cd16d

Update sentiment-analyser.py

Browse files
Files changed (1) hide show
  1. sentiment-analyser.py +1 -1
sentiment-analyser.py CHANGED
@@ -14,7 +14,7 @@ def classification(user_input, type):
14
  if type=="sentiment-analysis":
15
  classifier = pipeline("sentiment-analysis")
16
  elif type=="twitter-xlm-roberta-base-sentiment":
17
- classifier = pipeline("sentiment-analysis", model="cardiffnlp/twitter-xlm-roberta-base-sentiment")
18
  result = classifier(user_input)[0]
19
  return result
20
 
 
14
  if type=="sentiment-analysis":
15
  classifier = pipeline("sentiment-analysis")
16
  elif type=="twitter-xlm-roberta-base-sentiment":
17
+ classifier = pipeline("sentiment-analysis", model="cardiffnlp/twitter-xlm-roberta-base-sentiment", tokenizer="cardiffnlp/twitter-xlm-roberta-base-sentiment")
18
  result = classifier(user_input)[0]
19
  return result
20