try to fix ndcg bug

app.py

@@ -37,6 +37,36 @@ qrels = {}
 datasets = ["scifact"]
 current_dataset = "scifact"
 
+def log_system_info():
+    logger.info("System Information:")
+    logger.info(f"Python version: {sys.version}")
+    logger.info(f"Platform: {platform.platform()}")
+    
+    logger.info("\nPackage Versions:")
+    logger.info(f"torch: {torch.__version__}")
+    logger.info(f"transformers: {transformers.__version__}")
+    logger.info(f"peft: {peft.__version__}")
+    logger.info(f"faiss: {faiss.__version__}")
+    logger.info(f"gradio: {gradio.__version__}")
+    logger.info(f"pytrec_eval: {pytrec_eval.__version__}")
+    logger.info(f"ir_datasets: {ir_datasets.__version__}")
+    
+    if torch.cuda.is_available():
+        logger.info(f"\nCUDA Information:")
+        logger.info(f"CUDA available: Yes")
+        logger.info(f"CUDA version: {torch.version.cuda}")
+        logger.info(f"cuDNN version: {torch.backends.cudnn.version()}")
+        logger.info(f"Number of GPUs: {torch.cuda.device_count()}")
+        for i in range(torch.cuda.device_count()):
+            logger.info(f"GPU {i}: {torch.cuda.get_device_name(i)}")
+    else:
+        logger.info("\nCUDA Information:")
+        logger.info("CUDA available: No")
+
+
+log_system_info()
+
+
 
 def pool(last_hidden_states, attention_mask, pool_type="last"):
     last_hidden = last_hidden_states.masked_fill(~attention_mask[..., None].bool(), 0.0)