Spaces:

poltextlab
/

babel_machine

Running

poltextlab commited on 2 days ago

Commit

58f6b78

verified ·

1 Parent(s): 410b6ef

ver2

Files changed (1) hide show

interfaces/cap_media2.py CHANGED Viewed

@@ -31,7 +31,7 @@ def check_huggingface_path(checkpoint_path: str):
         return False
 def build_huggingface_path(language: str, domain: str):
-    return "poltextlab/xlm-roberta-large-pooled-cap-media2-v1"
 def predict(text, model_id, tokenizer_id):
     device = torch.device("cpu")

         return False
 def build_huggingface_path(language: str, domain: str):
+    return "poltextlab/xlm-roberta-large-pooled-cap-media2-v2"
 def predict(text, model_id, tokenizer_id):
     device = torch.device("cpu")