Update model.py

model.py

@@ -1,6 +1,8 @@
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import os
-
+import torch
+if torch.cuda.is_available():
+    model.to('cuda')
 def modelFeedback(ats_score, resume_data, job_description):
     """
     Generate ATS feedback by utilizing a pre-configured pipeline.
@@ -50,7 +52,6 @@ def modelFeedback(ats_score, resume_data, job_description):
             input_ids,
             max_length=1500,
             temperature=0.01,
-            top_p=0.7,
             pad_token_id=tokenizer.eos_token_id  # Ensure padding works properly
         )