poltextlab commited on
Commit
e1f54b1
1 Parent(s): 1600ed8

Update language capitalisation

Browse files
Files changed (1) hide show
  1. interfaces/sentiment.py +1 -1
interfaces/sentiment.py CHANGED
@@ -22,7 +22,7 @@ SENTIMENT_LABEL_NAMES = {0: "Negative", 1: "No sentiment or Neutral sentiment",
22
 
23
 
24
  def build_huggingface_path(language: str):
25
- if language == "czech" or language == "slovakian":
26
  return "visegradmedia-emotion/Emotion_RoBERTa_pooled_V4"
27
  return "poltextlab/xlm-roberta-large-pooled-MORES"
28
 
 
22
 
23
 
24
  def build_huggingface_path(language: str):
25
+ if language == "Czech" or language == "Slovakian":
26
  return "visegradmedia-emotion/Emotion_RoBERTa_pooled_V4"
27
  return "poltextlab/xlm-roberta-large-pooled-MORES"
28