add logs to print the local paths of the models

Author: thomaslim6793

indra/sources/indra_bert/api.py

@@ -1,6 +1,7 @@
 __all__ = ['process_text']
 
 import os
+from tqdm import tqdm
 import logging
 
 logger = logging.getLogger(__name__)
@@ -37,6 +38,9 @@ def create_extractor(
             role_model_path="thomaslim6793/indra_bert_indra_stmt_agents_role_assigner",
             stmt_conf_threshold=stmt_conf_threshold
         )
+    logger.info(f"Loaded ner_model from: {ise.ner_model_local_path}")
+    logger.info(f"Loaded stmt_model from: {ise.stmt_model_local_path}")
+    logger.info(f"Loaded role_model from: {ise.role_model_local_path}")
     return ise
 
 def process_text(text, 
@@ -53,7 +57,7 @@ def process_text(text,
     )
     res = ise.extract_structured_statements_batch(text)
     ip = IndraBertProcessor(res, grounder=grounder)
-    return ip
+    return ip, ise
 
 def process_texts(texts, 
                   ner_model_path="thomaslim6793/indra_bert_ner_agent_detection",
@@ -73,8 +77,8 @@ def process_texts(texts,
     )
 
     ips = []
-    for text in texts:
+    for text in tqdm(texts, desc="Processing texts"):
         res = ise.extract_structured_statements_batch(text)
         ip = IndraBertProcessor(res, grounder=grounder)
         ips.append(ip)
-    return ips
+    return ips, ise