Spaces:
Running
Running
kovacsvi
commited on
Commit
·
b272a72
1
Parent(s):
d73b37f
use the proper default path
Browse files- 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-
|
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")
|