Update model.py

model.py

@@ -1,22 +1,19 @@
-import google.generativeai as genai 
 import os
 from dotenv import load_dotenv
-from convert import ExtractPDFText
-load_dotenv()
-
-genai.configure(api_key = os.environ["GOOGLE_API_KEY"])
-
-model = genai.GenerativeModel("gemini-pro") 
+from transformers import pipeline
 
-def modelFeedback(ats_score,resume_data):
+load_dotenv()
+model_name = "gpt2" 
+generator = pipeline("text-generation", model=model_name)
 
+def modelFeedback(ats_score, resume_data):
     input_prompt = f"""
-        You are now an ATS Score analyzer and given ATS Score is {int(ats_score*100)}%. 
-        Your task is to provide feedback to the user based on the ATS score.
-        print ATS score first. mention where resume is good and where resume lacks. 
-        talk about each section of user's resume and talk good and bad points of it. 
+    You are now an ATS Score analyzer and given ATS Score is {int(ats_score * 100)}%. 
+    Your task is to provide feedback to the user based on the ATS score.
+    Print ATS score first. Mention where the resume is good and where the resume lacks. 
+    Talk about each section of the user's resume and talk about good and bad points of it. 
+    Resume Data: {resume_data}
     """
-    response = model.generate_content([input_prompt,resume_data],stream=True)
-    response.resolve()
-
+    response = generator(input_prompt, max_length=500, num_return_sequences=1)[0]['generated_text']
+    
     return response