Viper51 commited on
Commit
9b1cc6f
·
verified ·
1 Parent(s): f30680d

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -65,7 +65,7 @@ class evaluation(BaseModel):
65
  def get_llm(api_key):
66
  """Cached function to initialize the LLM."""
67
  return ChatGoogleGenerativeAI(
68
- model="gemini-2.5-flash",
69
  temperature=1.0,
70
  google_api_key=api_key # <-- Explicitly pass the key here
71
  )
 
65
  def get_llm(api_key):
66
  """Cached function to initialize the LLM."""
67
  return ChatGoogleGenerativeAI(
68
+ model="gemini-1.5-flash-latest",
69
  temperature=1.0,
70
  google_api_key=api_key # <-- Explicitly pass the key here
71
  )