Spaces:

poltextlab
/

babelmachine

Running

poltextlab commited on 7 days ago

Commit

cbde6a8

verified ·

1 Parent(s): 916ed1c

Slovakian ->Slovak

Files changed (1) hide show

interfaces/emotion.py CHANGED Viewed

@@ -12,14 +12,14 @@ from label_dicts import MANIFESTO_LABEL_NAMES
 HF_TOKEN = os.environ["hf_read"]
 languages = [
-    "Czech", "English", "French", "German", "Hungarian", "Polish", "Slovakian"
 ]
 domains = {
     "parliamentary speech": "parlspeech",
 }
 def build_huggingface_path(language: str):
-    if language == "Czech" or language == "Slovakian":
         return "visegradmedia-emotion/Emotion_RoBERTa_pooled_V4"
     return "poltextlab/xlm-roberta-large-pooled-MORES"

 HF_TOKEN = os.environ["hf_read"]
 languages = [
+    "Czech", "English", "French", "German", "Hungarian", "Polish", "Slovak"
 ]
 domains = {
     "parliamentary speech": "parlspeech",
 }
 def build_huggingface_path(language: str):
+    if language == "Czech" or language == "Slovak":
         return "visegradmedia-emotion/Emotion_RoBERTa_pooled_V4"
     return "poltextlab/xlm-roberta-large-pooled-MORES"