Spaces:
Sleeping
Sleeping
Update src/streamlit_app.py
Browse files- 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-
|
| 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 |
)
|