Update app.py

app.py

@@ -172,6 +172,8 @@ def show_mitigation(bias_type):
 
 #function to set up and return LLama 3.2-1b Instruct model transformer and its tokenizer + the Regard classifier model from Hugging Face
 def setup_models():
+    hf_token = os.getenv("fairLLM")
+    
     tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.2-1B-Instruct")
     #set pad_token_id to eos_token_id if it's not already defined
     if tokenizer.pad_token_id is None:
@@ -181,6 +183,7 @@ def setup_models():
     bias_pipe = pipeline(
         "text-generation",
         model="meta-llama/Llama-3.2-1B-Instruct",
+        token=hf_token,
         tokenizer=tokenizer, #pass the tokenizer with pad_token_id set
         device=0 if torch.cuda.is_available() else -1
     )