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)
|