Update model.py

model.py

@@ -22,8 +22,9 @@ def modelFeedback(ats_score, resume_data):
     input_ids = tokenizer(input_prompt, return_tensors="pt").input_ids
     input_length = input_ids.shape[1]
     
+    print(f"Input length: {input_length}")
 
-    # Generate response
-    response = generator(input_prompt, num_return_sequences=1)
+    # Generate response with max_new_tokens instead of max_length
+    response = generator(input_prompt, max_new_tokens=150, num_return_sequences=1)
 
-    return response
+    return response[0]['generated_text']