Spaces:
Running
Running
Add slovakian and czech languages with the appropriate model
Browse files- interfaces/emotion.py +3 -1
interfaces/emotion.py
CHANGED
|
@@ -12,13 +12,15 @@ 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",
|
| 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):
|
|
|
|
| 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",
|
| 19 |
}
|
| 20 |
|
| 21 |
def build_huggingface_path(language: str):
|
| 22 |
+
if language == "czech" or language == "slovakian":
|
| 23 |
+
return "visegradmedia-emotion/Emotion_RoBERTa_pooled_V4"
|
| 24 |
return "poltextlab/xlm-roberta-large-pooled-MORES"
|
| 25 |
|
| 26 |
def predict(text, model_id, tokenizer_id):
|