kovacsvi commited on
Commit
444ac6d
·
1 Parent(s): f295362

use twitter model for social media

Browse files
Files changed (1) hide show
  1. 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):