Spaces:
Running
Running
kovacsvi
commited on
Commit
·
444ac6d
1
Parent(s):
f295362
use twitter model for social media
Browse files- interfaces/cap_minor.py +2 -0
interfaces/cap_minor.py
CHANGED
@@ -63,6 +63,8 @@ def check_huggingface_path(checkpoint_path: str):
|
|
63 |
return False
|
64 |
|
65 |
def build_huggingface_path(language: str, domain: str):
|
|
|
|
|
66 |
return "poltextlab/xlm-roberta-large-pooled-cap-minor-v3"
|
67 |
|
68 |
def predict(text, model_id, tokenizer_id):
|
|
|
63 |
return False
|
64 |
|
65 |
def build_huggingface_path(language: str, domain: str):
|
66 |
+
if domain in ["social"]:
|
67 |
+
return "poltextlab/xlm-roberta-large-twitter-cap-minor"
|
68 |
return "poltextlab/xlm-roberta-large-pooled-cap-minor-v3"
|
69 |
|
70 |
def predict(text, model_id, tokenizer_id):
|