crcdng commited on
Commit
1651306
·
1 Parent(s): e9a8597

InferenceClientModel

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
@@ -24,7 +24,7 @@ def get_current_time_in_timezone(timezone: str) -> str:
24
 
25
  final_answer = FinalAnswerTool()
26
 
27
- model = HfApiModel(
28
  max_tokens=2096,
29
  temperature=0.5,
30
  model_id="Qwen/Qwen2.5-Coder-32B-Instruct", # it is possible that this model may be overloaded
 
1
+ from smolagents import CodeAgent, DuckDuckGoSearchTool, InferenceClientModel, load_tool, tool
2
  import datetime
3
  import requests
4
  import pytz
 
24
 
25
  final_answer = FinalAnswerTool()
26
 
27
+ model = InferenceClientModel(
28
  max_tokens=2096,
29
  temperature=0.5,
30
  model_id="Qwen/Qwen2.5-Coder-32B-Instruct", # it is possible that this model may be overloaded