Spaces:
Running
Running
Slovakian -> Slovak
Browse files- interfaces/sentiment.py +2 -2
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 |
-
"Czech", "English", "French", "German", "Hungarian", "Polish", "
|
16 |
]
|
17 |
domains = {
|
18 |
"parliamentary speech": "parlspeech",
|
@@ -22,7 +22,7 @@ SENTIMENT_LABEL_NAMES = {0: "Negative", 1: "No sentiment or Neutral sentiment",
|
|
22 |
|
23 |
|
24 |
def build_huggingface_path(language: str):
|
25 |
-
if language == "Czech" or language == "
|
26 |
return "visegradmedia-emotion/Emotion_RoBERTa_pooled_V4"
|
27 |
return "poltextlab/xlm-roberta-large-pooled-MORES"
|
28 |
|
|
|
12 |
HF_TOKEN = os.environ["hf_read"]
|
13 |
|
14 |
languages = [
|
15 |
+
"Czech", "English", "French", "German", "Hungarian", "Polish", "Slovak"
|
16 |
]
|
17 |
domains = {
|
18 |
"parliamentary speech": "parlspeech",
|
|
|
22 |
|
23 |
|
24 |
def build_huggingface_path(language: str):
|
25 |
+
if language == "Czech" or language == "Slovak":
|
26 |
return "visegradmedia-emotion/Emotion_RoBERTa_pooled_V4"
|
27 |
return "poltextlab/xlm-roberta-large-pooled-MORES"
|
28 |
|