Spaces:

Krish30
/

final-yr-cv-shortlisting-rag-ai

Sleeping

Krish30 commited on Feb 27

Commit

84746f8

verified ·

1 Parent(s): 3033d33

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -38,7 +38,7 @@ def retrieve_from_vectorstore(vectorstore, query):
 # Get response from Generative AI
 def get_gemini_response(prompt):
-    model = genai.GenerativeModel('gemini-pro')
     response = model.generate_content(prompt)
     return response.candidates[0].content.parts[0].text if response else None

 # Get response from Generative AI
 def get_gemini_response(prompt):
+    model = genai.GenerativeModel('gemini-2.0-flash')
     response = model.generate_content(prompt)
     return response.candidates[0].content.parts[0].text if response else None