Hugging Face's logo

Spaces:
Kackle
/
Final_Assignment_Template
Sleeping

App Files Community
Kackle commited on
Commit
e4c97f8
·
verified ·
1 Parent(s): 5216f08

v

Browse files
Files changed (1) hide show
  1. gemini_agent.py +1 -1
gemini_agent.py CHANGED
@@ -14,7 +14,7 @@ class GeminiAgent:
14
  api_key = os.getenv('GOOGLE_API_KEY')
15
  genai.configure(api_key=api_key)
16
 
17
- self.model = genai.GenerativeModel('gemini-1.5-flash-8b')
18
 
19
  # Initialize parsers
20
  self.excel_parser = ExcelParser()
 
14
  api_key = os.getenv('GOOGLE_API_KEY')
15
  genai.configure(api_key=api_key)
16
 
17
+ self.model = genai.GenerativeModel('gemini-2.0-flash-exp')
18
 
19
  # Initialize parsers
20
  self.excel_parser = ExcelParser()