bainskarman commited on
Commit
b2a231b
·
verified ·
1 Parent(s): cda75bc

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -3
model.py CHANGED
@@ -29,9 +29,7 @@ def modelFeedback(ats_score, resume_data, job_description):
29
  #### Job Description: {job_description}
30
  """
31
 
32
- # Load the pipeline
33
- pipe = pipeline("text-generation", model="meta-llama/Llama-3.2-3B-Instruct")
34
-
35
  # Generate the response using the pipeline
36
  try:
37
  output = pipe(input_prompt, max_length=1500, temperature=0.01, top_p=0.7, return_full_text=True)
 
29
  #### Job Description: {job_description}
30
  """
31
 
32
+ pipe = pipeline("text-generation", model="meta-llama/Llama-3.2-1B")
 
 
33
  # Generate the response using the pipeline
34
  try:
35
  output = pipe(input_prompt, max_length=1500, temperature=0.01, top_p=0.7, return_full_text=True)