Update mmlu_pro_eval_adapted.py

mmlu_pro_eval_adapted.py

@@ -195,6 +195,8 @@ def eval_cot(subject, model, tokenizer, val_df, test_df, num_shots=5):
 
 @spaces.GPU(duration=240)  # Extended to 3 minutes for larger evaluations
 def evaluate_mmlu_pro(model_name, num_subjects=-1, num_questions=10, num_shots=5):
+    print ("IS CUDA AVAILABLE: ", torch.cuda.is_available())
+    
     model, tokenizer = load_model(model_name, gpu_utilization=0.8)
     
     # Ensure model is in evaluation mode