Files changed (1) hide show
  1. app.py +10 -2
app.py CHANGED
@@ -1,4 +1,5 @@
1
- from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
 
2
  import datetime
3
  import requests
4
  import pytz
@@ -39,7 +40,14 @@ final_answer = FinalAnswerTool()
39
  # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
40
  # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
41
 
42
- model = HfApiModel(
 
 
 
 
 
 
 
43
  max_tokens=2096,
44
  temperature=0.5,
45
  model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
 
1
+ #from smolagents import CodeAgent, DuckDuckGoSearchTool, FinalAnswerTool, HfApiModel,load_tool,tool
2
+ from smolagents import CodeAgent, DuckDuckGoSearchTool, InferenceClientModel, load_tool, tool
3
  import datetime
4
  import requests
5
  import pytz
 
40
  # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
41
  # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'
42
 
43
+ #model = HfApiModel(
44
+ #max_tokens=2096,
45
+ #temperature=0.5,
46
+ #model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
47
+ #custom_role_conversions=None,
48
+ #)
49
+
50
+ model = InferenceClientModel(
51
  max_tokens=2096,
52
  temperature=0.5,
53
  model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded