Update model.py

model.py

@@ -1,5 +1,9 @@
 import torch
 from transformers import pipeline
+import os
+
+# Retrieve the token stored as a secret in the environment
+hf_token = os.getenv('KEY2')
 
 # Define model id
 model_id = "meta-llama/Llama-3.2-1B"
@@ -9,7 +13,8 @@ pipe = pipeline(
     "text-generation", 
     model=model_id, 
     torch_dtype=torch.bfloat16, 
-    device_map="auto"
+    device_map="auto", 
+    use_auth_token=hf_token
 )
 
 def modelFeedback(ats_score, resume_data, job_description):