sm2899 commited on
Commit
d454322
1 Parent(s): bfde37d

Update sentiment-analyser.py

Browse files
Files changed (1) hide show
  1. sentiment-analyser.py +1 -1
sentiment-analyser.py CHANGED
@@ -16,7 +16,7 @@ def classification(user_input, type):
16
  elif type=="twitter-xlm-roberta-base-sentiment":
17
  path="cardiffnlp/twitter-xlm-roberta-base-sentiment"
18
  classifier = pipeline("sentiment-analysis", model=path, tokenizer=path)
19
- elif type=="sentiment-roberta-large-english-3-classes":
20
  path="siebert/sentiment-roberta-large-english"
21
  classifier = pipeline("sentiment-analysis", model=path)
22
  result = classifier(user_input)
 
16
  elif type=="twitter-xlm-roberta-base-sentiment":
17
  path="cardiffnlp/twitter-xlm-roberta-base-sentiment"
18
  classifier = pipeline("sentiment-analysis", model=path, tokenizer=path)
19
+ elif type=="sentiment-roberta-large-english":
20
  path="siebert/sentiment-roberta-large-english"
21
  classifier = pipeline("sentiment-analysis", model=path)
22
  result = classifier(user_input)