poltextlab commited on
Commit
06fb9d6
1 Parent(s): 25980af

update emotions model from xlm-roberta-large-pooled-emotions to xlm-roberta-large-pooled-MORES

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
- return "poltextlab/xlm-roberta-large-pooled-emotions"
23
 
24
  def predict(text, model_id, tokenizer_id):
25
  device = torch.device("cpu")
 
19
  }
20
 
21
  def build_huggingface_path(language: str):
22
+ return "poltextlab/xlm-roberta-large-pooled-MORES"
23
 
24
  def predict(text, model_id, tokenizer_id):
25
  device = torch.device("cpu")