poltextlab commited on
Commit
6e3bd95
1 Parent(s): 501e4bf

Add slovakian and czech language with the apropriate model

Browse files
Files changed (1) hide show
  1. interfaces/sentiment.py +3 -1
interfaces/sentiment.py CHANGED
@@ -12,7 +12,7 @@ from label_dicts import MANIFESTO_LABEL_NAMES
12
  HF_TOKEN = os.environ["hf_read"]
13
 
14
  languages = [
15
- "English", "French", "German", "Hungarian", "Polish"
16
  ]
17
  domains = {
18
  "parliamentary speech": "parlspeech",
@@ -22,6 +22,8 @@ SENTIMENT_LABEL_NAMES = {0: "Negative", 1: "No sentiment or Neutral sentiment",
22
 
23
 
24
  def build_huggingface_path(language: str):
 
 
25
  return "poltextlab/xlm-roberta-large-pooled-MORES"
26
 
27
  def predict(text, model_id, tokenizer_id):
 
12
  HF_TOKEN = os.environ["hf_read"]
13
 
14
  languages = [
15
+ "Czech", "English", "French", "German", "Hungarian", "Polish", "Slovakian"
16
  ]
17
  domains = {
18
  "parliamentary speech": "parlspeech",
 
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
 
29
  def predict(text, model_id, tokenizer_id):