Spaces:
Running
Running
minor v2
Browse files- 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")
|