Imarticuslearning commited on
Commit
1294c01
·
verified ·
1 Parent(s): ef28f5a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -724,7 +724,7 @@ if st.session_state["selected_domain"] == "Soft Skills":
724
  st.warning("⚠️ Please upload a resume.")
725
  st.stop()
726
  prompt = f"Ask {num_qs} HR-style interview questions based on this resume: {input_text}"
727
- model = genai.GenerativeModel('gemini-1.5-pro-latest')
728
  response = model.generate_content([prompt])
729
  questions = [q.strip("* ") for q in response.text.strip().split("\n") if q.strip()]
730
  st.session_state["generated_questions"] = questions[:num_qs]
@@ -865,7 +865,7 @@ else:
865
  st.session_state["generated_questions"] = sample(questions_from_excel, min(num_qs, len(questions_from_excel)))
866
  else:
867
  prompt = f"Ask {num_qs} direct and core-level {difficulty} interview questions related to {input_text}. Do not include intros or numbering."
868
- model = genai.GenerativeModel('gemini-1.5-pro-latest')
869
  response = model.generate_content([prompt, input_text])
870
  lines = response.text.strip().split("\n")
871
  questions = [q.strip("* ") for q in lines if q.strip()]
 
724
  st.warning("⚠️ Please upload a resume.")
725
  st.stop()
726
  prompt = f"Ask {num_qs} HR-style interview questions based on this resume: {input_text}"
727
+ model = genai.GenerativeModel('gemini-2.0-flash-lite')
728
  response = model.generate_content([prompt])
729
  questions = [q.strip("* ") for q in response.text.strip().split("\n") if q.strip()]
730
  st.session_state["generated_questions"] = questions[:num_qs]
 
865
  st.session_state["generated_questions"] = sample(questions_from_excel, min(num_qs, len(questions_from_excel)))
866
  else:
867
  prompt = f"Ask {num_qs} direct and core-level {difficulty} interview questions related to {input_text}. Do not include intros or numbering."
868
+ model = genai.GenerativeModel('gemini-2.0-flash-lite')
869
  response = model.generate_content([prompt, input_text])
870
  lines = response.text.strip().split("\n")
871
  questions = [q.strip("* ") for q in lines if q.strip()]