| import os | |
| from huggingface_hub import login, HfApi | |
| # Authenticate with Hugging Face using the token from Space secrets | |
| hf_token = os.getenv("HF_TOKEN") | |
| if hf_token: | |
| login(token=hf_token) | |
| print("Authenticated with Hugging Face token.") | |
| else: | |
| print("HF_TOKEN not found in environment variables. Please set it in Space secrets.") | |
| # Test access to the gated repo | |
| try: | |
| api = HfApi() | |
| model_info = api.model_info("meta-llama/Llama-2-7b-hf") | |
| print("Successfully accessed LLaMA-2-7b-hf repo:", model_info.id) | |
| except Exception as e: | |
| print(f"Error accessing gated repo: {e}") |