Update hf_guardrail.py

hf_guardrail.py

@@ -1,9 +1,12 @@
 from transformers import pipeline
 import os
+from dotenv import load_dotenv
+
+load_dotenv()
+hf_token = os.getenv('HF_TOKEN')
 
 # Initialize Hugging Face pipeline for text classification or question-answering task
 model_name = "meta-llama/Meta-Llama-Guard-3-8B"  # Use a safety guard model or choose a suitable Hugging Face model
-hf_token = os.getenv('HF_TOKEN')
 
 # Load the Hugging Face model and tokenizer for safety checking (using Hugging Face's `pipeline` API)
 safety_model = pipeline('text-generation', model=model_name, use_auth_token=hf_token)