thewebwelost commited on
Commit
c95a627
·
verified ·
1 Parent(s): cb35c65

Update app.py

Browse files

try LiteLLMModel

Files changed (1) hide show
  1. app.py +2 -7
app.py CHANGED
@@ -1,4 +1,4 @@
1
- from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
2
  import datetime
3
  import requests
4
  import pytz
@@ -35,12 +35,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
35
 
36
  final_answer = FinalAnswerTool()
37
 
38
- model = HfApiModel(
39
- max_tokens=2096,
40
- temperature=0.5,
41
- model_id=' https://jc26mwg228mkj8dw.us-east-1.aws.endpoints.huggingface.cloud',# it is possible that this model may be overloaded
42
- custom_role_conversions=None,
43
- )
44
 
45
 
46
  # Import tool from Hub
 
1
+ from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel, load_tool, tool, LiteLLMModel
2
  import datetime
3
  import requests
4
  import pytz
 
35
 
36
  final_answer = FinalAnswerTool()
37
 
38
+ model = LiteLLMModel(model_id="gemini/gemini-2.0-flash")
 
 
 
 
 
39
 
40
 
41
  # Import tool from Hub