Spaces:
Sleeping
Sleeping
Commit
·
c714b68
1
Parent(s):
a5d56b4
Some changes for gemini 2.5 flash lite
Browse files
app.py
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
-
from smolagents import CodeAgent,
|
2 |
-
from smolagents import LiteLLMModel # For
|
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.
|
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)
|