Update model.py

model.py

@@ -1,4 +1,4 @@
-from transformers import AutoTokenizer, AutoModelForCausalLM
+from transformers import AutoTokenizer, AutoModelForCausalLM, AutoModelForCausalLM
 import os
 import torch
 
@@ -8,15 +8,11 @@ device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
 # Load the tokenizer and model once, outside of the function
 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
-).to(device)
+
+# Load the tokenizer and model directly
+tokenizer = AutoTokenizer.from_pretrained("deepseek-ai/DeepSeek-R1", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("deepseek-ai/DeepSeek-R1", trust_remote_code=True)
 
 def modelFeedback(ats_score, resume_data, job_description):
     """