Update model.py

model.py

@@ -1,10 +1,13 @@
 import os
 from dotenv import load_dotenv
-from transformers import pipeline
+from transformers import pipeline, AutoTokenizer
 
 load_dotenv()
-model_name = "gpt2" 
+
+# Load a larger Hugging Face model
+model_name = "EleutherAI/gpt-neo-2.7B"
 generator = pipeline("text-generation", model=model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
 
 def modelFeedback(ats_score, resume_data):
     input_prompt = f"""
@@ -14,6 +17,20 @@ def modelFeedback(ats_score, resume_data):
     Talk about each section of the user's resume and talk about good and bad points of it. 
     Resume Data: {resume_data}
     """
-    response = generator(input_prompt, max_new_tokens=150, num_return_sequences=1)[0]['generated_text']
     
+    # Tokenize the input to check its length
+    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, max_new_tokens=150, num_return_sequences=1)
+    
+    # Check if response is not empty
+    if response and len(response) > 0:
+        generated_text = response[0]['generated_text']
+    else:
+        generated_text = "No response generated."
+    response = generated_text
     return response