Spaces:
Running
Running
buggy model_id
Browse files- 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"
|