APRG commited on
Commit
6d55a00
·
verified ·
1 Parent(s): 71c08fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ class BasicAgent:
17
  print("BasicAgent initialized.")
18
  # Initialize the Hugging Face model
19
  llm = LiteLLMModel(
20
- model_id="gemini-2.0-flash", api_key=os.getenv("GEMINI_API_KEY")
21
  )
22
 
23
  # Initialize tools
 
17
  print("BasicAgent initialized.")
18
  # Initialize the Hugging Face model
19
  llm = LiteLLMModel(
20
+ model_id="gemini-2.0-flash", api_key=os.getenv("GOOGLE_API_KEY")
21
  )
22
 
23
  # Initialize tools