poltextlab commited on
Commit
eea7116
·
verified ·
1 Parent(s): a5ad33c
Files changed (1) hide show
  1. interfaces/cap_minor.py +1 -1
interfaces/cap_minor.py CHANGED
@@ -41,7 +41,7 @@ def check_huggingface_path(checkpoint_path: str):
41
  return False
42
 
43
  def build_huggingface_path(language: str, domain: str):
44
- return "poltextlab/xlm-roberta-large-pooled-cap-minor"
45
 
46
  def predict(text, model_id, tokenizer_id):
47
  device = torch.device("cpu")
 
41
  return False
42
 
43
  def build_huggingface_path(language: str, domain: str):
44
+ return "poltextlab/xlm-roberta-large-pooled-cap-minor-v2"
45
 
46
  def predict(text, model_id, tokenizer_id):
47
  device = torch.device("cpu")