add log

pages/summarization_playground.py

@@ -8,6 +8,8 @@ from utils.data import dataset
 import gc
 import torch
 
+import logging
+
 load_dotenv()
 
 __model_on_gpu__ = ''
@@ -37,6 +39,7 @@ def generate_answer(sources, model_name, prompt):
 
     if __model_on_gpu__ != model_name:
         if __model_on_gpu__:
+            logging.info(f"delete model {__model_on_gpu__}")
             del model[__model_on_gpu__]
             gc.collect()
             torch.cuda.empty_cache()

utils/model.py

@@ -5,6 +5,8 @@ import torch
 from huggingface_hub import login
 import os 
 
+import logging
+
 login(token = os.getenv('HF_TOKEN'))
 
 class Model(torch.nn.Module):
@@ -28,6 +30,7 @@ class Model(torch.nn.Module):
             device_map="auto",
         )
 
+        logging.info(f'Load model {self.name}')
         self.update()
 
     @classmethod