Spaces:

poltextlab
/

babel_machine

Running

kovacsvi commited on May 7

Commit

2ae6568

1 Parent(s): fcc4623

added tmp

Files changed (1) hide show

interfaces/cap_minor_media.py CHANGED Viewed

@@ -128,7 +128,7 @@ def predict(text, major_model_id, minor_model_id, tokenizer_id, HF_TOKEN=None):
     return output_pred, output_info
-def predict_cap(method, text, language, domain):
     domain = domains[domain]
     major_model_id, minor_model_id = build_huggingface_path(language, domain)
     tokenizer_id = "xlm-roberta-large"

     return output_pred, output_info
+def predict_cap(tmp, method, text, language, domain):
     domain = domains[domain]
     major_model_id, minor_model_id = build_huggingface_path(language, domain)
     tokenizer_id = "xlm-roberta-large"