Update database.py

database.py

@@ -8,10 +8,11 @@ from datasets import Dataset, load_dataset
 from transformers import AutoTokenizer, AutoModel
 import torch
 
+
 # User-configurable variables
 DB_NAME = "python_programs"  # ChromaDB collection name
 HF_DATASET_NAME = "python_program_vectors"  # Hugging Face Dataset name
-HF_KEY = "HF_KEY"  # Replace with your Hugging Face API token
+HF_KEY = os.getenv("HF_KEY")  # Replace with your Hugging Face API token
 PERSIST_DIR = "./chroma_data"  # Directory for persistent storage (optional)
 USE_GPU = False  # Default to CPU, set to True for GPU if available