poltextlab commited on
Commit
8dc5af0
1 Parent(s): a7a15af

buggy model_id

Browse files
Files changed (1) hide show
  1. interfaces/cap.py +1 -0
interfaces/cap.py CHANGED
@@ -52,6 +52,7 @@ def check_huggingface_path(checkpoint_path: str):
52
  return False
53
 
54
  def build_huggingface_path(language: str, domain: str):
 
55
  base_path = "xlm-roberta-large"
56
  lang_domain_path = f"poltextlab/{base_path}-{language}-{domain}-cap-v3"
57
  lang_path = f"poltextlab/{base_path}-{language}-cap-v3"
 
52
  return False
53
 
54
  def build_huggingface_path(language: str, domain: str):
55
+ language = language.lower()
56
  base_path = "xlm-roberta-large"
57
  lang_domain_path = f"poltextlab/{base_path}-{language}-{domain}-cap-v3"
58
  lang_path = f"poltextlab/{base_path}-{language}-cap-v3"