kovacsvi commited on
Commit
b272a72
·
1 Parent(s): d73b37f

use the proper default path

Browse files
Files changed (1) hide show
  1. interfaces/emotion.py +1 -1
interfaces/emotion.py CHANGED
@@ -23,7 +23,7 @@ domains = {
23
  def build_huggingface_path(language: str):
24
  if language == "Czech" or language == "Slovak":
25
  return "visegradmedia-emotion/Emotion_RoBERTa_pooled_V4"
26
- return "poltextlab/xlm-roberta-large-pooled-MORES"
27
 
28
  def predict(text, model_id, tokenizer_id):
29
  device = torch.device("cpu")
 
23
  def build_huggingface_path(language: str):
24
  if language == "Czech" or language == "Slovak":
25
  return "visegradmedia-emotion/Emotion_RoBERTa_pooled_V4"
26
+ return "poltextlab/xlm-roberta-large-pooled-emotions6"
27
 
28
  def predict(text, model_id, tokenizer_id):
29
  device = torch.device("cpu")