davemasino commited on
Commit
c714b68
·
1 Parent(s): a5d56b4

Some changes for gemini 2.5 flash lite

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -1,5 +1,5 @@
1
- from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
2
- from smolagents import LiteLLMModel # For Gemeni
3
  import os
4
  import datetime
5
  import requests
@@ -61,7 +61,7 @@ web_search = DuckDuckGoSearchTool()
61
  #custom_role_conversions=None,
62
  #)
63
 
64
- model = LiteLLMModel(model_id="gemini/gemini-2.0-flash-lite", api_key=os.getenv(key="GEMINI_API_KEY"))
65
 
66
  # Import tool from Hub
67
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
 
1
+ from smolagents import CodeAgent, uckDuckGoSearchTool, HfApiModel, load_tool, tool
2
+ from smolagents import LiteLLMModel # For Gemini 2.5
3
  import os
4
  import datetime
5
  import requests
 
61
  #custom_role_conversions=None,
62
  #)
63
 
64
+ model = LiteLLMModel(model_id="gemini/gemini-2.5-flash-lite-preview-06-17", api_key=os.getenv(key="GEMINI_API_KEY"))
65
 
66
  # Import tool from Hub
67
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)