david-thrower commited on
Commit
4cae411
·
verified ·
1 Parent(s): c001a98

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +48 -0
app.py ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ from smolagents import (
3
+ TransformersModel,
4
+ # load_tool,
5
+ DuckDuckGoSearchTool,
6
+ # ApiWebSearchTool,
7
+ VisitWebpageTool,
8
+ # WebSearchTool,
9
+ WikipediaSearchTool,
10
+ PythonInterpreterTool,
11
+ UserInputTool,
12
+ FinalAnswerTool,
13
+ CodeAgent,
14
+ InferenceClientModel,
15
+ GradioUI
16
+ )
17
+
18
+
19
+ MODEL_NAME = "Qwen/Qwen3-1.7B"
20
+
21
+
22
+ model =\
23
+ TransformersModel(
24
+ model_id=MODEL_NAME,
25
+ max_new_tokens=32768,
26
+ temperature=0.6,
27
+ top_p=0.95,
28
+ top_k=20,
29
+ repetition_penalty=1.1
30
+ )
31
+
32
+
33
+ agent = CodeAgent(
34
+ tools=[ DuckDuckGoSearchTool(),
35
+ # ApiWebSearchTool(),
36
+ VisitWebpageTool(),
37
+ # WebSearchTool(),
38
+ WikipediaSearchTool(),
39
+ PythonInterpreterTool(),
40
+ UserInputTool(),
41
+ FinalAnswerTool()
42
+
43
+ ],
44
+ model=model,
45
+ )
46
+
47
+
48
+ GradioUI(agent).launch()