samwill89 commited on
Commit
48ccd1e
·
verified ·
1 Parent(s): c7d4dcb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -3,9 +3,9 @@
3
  from smolagents import (
4
  CodeAgent,
5
  DuckDuckGoSearchTool,
6
- InferenceClientModel,
7
  load_tool,
8
  tool,
 
9
  )
10
  import datetime
11
  import re
@@ -67,11 +67,10 @@ def echo_and_length(text: str) -> str:
67
 
68
  final_answer = FinalAnswerTool()
69
 
70
- model = InferenceClientModel(
71
- max_tokens=2096,
72
- temperature=0.5,
73
  model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
74
- custom_role_conversions=None,
 
75
  )
76
 
77
  # Import a community tool from the Hub (text-to-image)
 
3
  from smolagents import (
4
  CodeAgent,
5
  DuckDuckGoSearchTool,
 
6
  load_tool,
7
  tool,
8
+ HfApiModel,
9
  )
10
  import datetime
11
  import re
 
67
 
68
  final_answer = FinalAnswerTool()
69
 
70
+ model = HfApiModel(
 
 
71
  model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
72
+ temperature=0.5,
73
+ max_tokens=2096,
74
  )
75
 
76
  # Import a community tool from the Hub (text-to-image)