sm2899 commited on
Commit
189f2e0
·
1 Parent(s): aa3328c

Update sentiment-analyser.py

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