Sripriya16 commited on
Commit
daf4097
·
verified ·
1 Parent(s): ab65ffa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -134,7 +134,7 @@ def generate_structured_json(text_to_analyze):
134
 
135
  def check_relevance_with_gemini(summary_text):
136
  """Checks if the summary is relevant to KMRL."""
137
- model = genai.GenerativeModel('gemini-1.5-flash-latest')
138
  prompt = f'Is this summary related to transportation, infrastructure, railways, or metro systems? Answer only "Yes" or "No".\n\nSummary: {summary_text}'
139
  response = model.generate_content(prompt)
140
  return "yes" in response.text.strip().lower()
 
134
 
135
  def check_relevance_with_gemini(summary_text):
136
  """Checks if the summary is relevant to KMRL."""
137
+ model = genai.GenerativeModel('gemini-2.5-flash')
138
  prompt = f'Is this summary related to transportation, infrastructure, railways, or metro systems? Answer only "Yes" or "No".\n\nSummary: {summary_text}'
139
  response = model.generate_content(prompt)
140
  return "yes" in response.text.strip().lower()