bainskarman commited on
Commit
73ae512
·
verified ·
1 Parent(s): 672bc9d

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -14,6 +14,6 @@ def modelFeedback(ats_score, resume_data):
14
  Talk about each section of the user's resume and talk about good and bad points of it.
15
  Resume Data: {resume_data}
16
  """
17
- response = generator(input_prompt, max_length=500, num_return_sequences=1)[0]['generated_text']
18
 
19
  return response
 
14
  Talk about each section of the user's resume and talk about good and bad points of it.
15
  Resume Data: {resume_data}
16
  """
17
+ response = generator(input_prompt, max_new_tokens=150, num_return_sequences=1)[0]['generated_text']
18
 
19
  return response