Update model.py

model.py

@@ -1,6 +1,8 @@
 from transformers import pipeline
 from transformers import AutoTokenizer, AutoModelForCausalLM
+import os
 
+)
 def modelFeedback(ats_score, resume_data, job_description):
     """
     Generate ATS feedback by utilizing a pre-configured pipeline.
@@ -32,6 +34,15 @@ def modelFeedback(ats_score, resume_data, job_description):
 
 
     # Load the tokenizer and model
+    huggingface_token = os.environ.get("KEY2")
+
+    tokenizer = AutoTokenizer.from_pretrained(
+        "meta-llama/Llama-3.2-1B",
+        use_auth_token=huggingface_token
+    )
+    model = AutoModelForCausalLM.from_pretrained(
+        "meta-llama/Llama-3.2-1B",
+        use_auth_token=huggingface_token
     tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.2-1B")
     model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-3.2-1B")
     try: