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

update language check to capitalised

Browse files
Files changed (1) hide show
  1. interfaces/emotion.py +1 -1
interfaces/emotion.py CHANGED
@@ -19,7 +19,7 @@ domains = {
19
  }
20
 
21
  def build_huggingface_path(language: str):
22
- if language == "czech" or language == "slovakian":
23
  return "visegradmedia-emotion/Emotion_RoBERTa_pooled_V4"
24
  return "poltextlab/xlm-roberta-large-pooled-MORES"
25
 
 
19
  }
20
 
21
  def build_huggingface_path(language: str):
22
+ if language == "Czech" or language == "Slovakian":
23
  return "visegradmedia-emotion/Emotion_RoBERTa_pooled_V4"
24
  return "poltextlab/xlm-roberta-large-pooled-MORES"
25