logging

app.py

@@ -23,7 +23,11 @@ from collections import defaultdict
 set_seed(42)
 
 # Set up logging
-logging.basicConfig(level=logging.INFO)
+# Set up logging with time printing
+logging.basicConfig(
+    format='%(asctime)s %(levelname)-8s %(message)s',
+    level=logging.INFO,
+    datefmt='%Y-%m-%d %H:%M:%S')
 logger = logging.getLogger(__name__)
 
 # Authenticate with HF_TOKEN
@@ -130,7 +134,7 @@ class RepLlamaModel:
         # if model is not on cuda, put it there
         if self.model.device.type != "cuda":
             self.model = self.model.cuda()
-            
+
         all_embeddings = []
         for i in tqdm.tqdm(range(0, len(texts), batch_size)):
             batch_texts = texts[i:i+batch_size]