Update src/litbench_pipeline.py

src/litbench_pipeline.py

@@ -34,14 +34,14 @@ from peft.peft_model import PeftModel
 from datasets import load_dataset
 from huggingface_hub import login
 
+
 # Load token from Hugging Face Secrets
-HF_TOKEN = os.getenv("HF_TOKEN")
-# Login using the token (only needed if loading private models)
-print("HF_TOKEN: ", HF_TOKEN)
+HF_TOKEN = os.getenv("access_token")
+# Login using the token
 if HF_TOKEN:
-    print("HERE???")
     login(HF_TOKEN)
 
+
 # Function to determine the chatbot's first message based on user choices
 def setup(download_option, train_option):
     download_papers.value = (download_option == "Download Paper")