AlexKurian commited on
Commit
98060ad
·
1 Parent(s): 3c54c83

Added gemini

Browse files
Files changed (1) hide show
  1. explainability.py +1 -1
explainability.py CHANGED
@@ -28,7 +28,7 @@ class ExplanationGenerator:
28
  # api_key=config.GROQ_API_KEY
29
  # )
30
  self.llm = ChatGoogleGenerativeAI(
31
- model="gemini-1.5-flash",
32
  temperature=0.5,
33
  google_api_key=config.GEMINI_API_KEY
34
  )
 
28
  # api_key=config.GROQ_API_KEY
29
  # )
30
  self.llm = ChatGoogleGenerativeAI(
31
+ model="gemini-2.5-flash",
32
  temperature=0.5,
33
  google_api_key=config.GEMINI_API_KEY
34
  )