Imm0rt4l commited on
Commit
afb3a0c
verified
1 Parent(s): 3cb9499

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +41 -41
app.py CHANGED
@@ -1,41 +1,41 @@
1
- import yaml
2
- import os
3
- from smolagents import GradioUI, CodeAgent, InferenceClientModel
4
-
5
- # Get current directory path
6
- CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
7
-
8
- from tools.web_search import DuckDuckGoSearchTool as WebSearch
9
- from tools.final_answer import FinalAnswerTool as FinalAnswer
10
-
11
-
12
-
13
- model = InferenceClientModel(
14
- model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
15
- )
16
-
17
- web_search = WebSearch()
18
- final_answer = FinalAnswer()
19
-
20
-
21
- with open(os.path.join(CURRENT_DIR, "prompts.yaml"), 'r') as stream:
22
- prompt_templates = yaml.safe_load(stream)
23
-
24
- agent = CodeAgent(
25
- model=model,
26
- tools=[web_search],
27
- managed_agents=[],
28
- class='CodeAgent',
29
- max_steps=20,
30
- verbosity_level=1,
31
- grammar=None,
32
- planning_interval=None,
33
- name=None,
34
- description=None,
35
- executor_type='local',
36
- executor_kwargs={},
37
- max_print_outputs_length=None,
38
- prompt_templates=prompt_templates
39
- )
40
- if __name__ == "__main__":
41
- GradioUI(agent).launch()
 
1
+ import yaml
2
+ import os
3
+ from smolagents import GradioUI, CodeAgent, InferenceClientModel
4
+
5
+ # Get current directory path
6
+ CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
7
+
8
+ from tools.web_search import DuckDuckGoSearchTool as WebSearch
9
+ from tools.final_answer import FinalAnswerTool as FinalAnswer
10
+
11
+
12
+
13
+ model = InferenceClientModel(
14
+ model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
15
+ )
16
+
17
+ web_search = WebSearch()
18
+ final_answer = FinalAnswer()
19
+
20
+
21
+ with open(os.path.join(CURRENT_DIR, "prompts.yaml"), 'r') as stream:
22
+ prompt_templates = yaml.safe_load(stream)
23
+
24
+ agent = CodeAgent(
25
+ model=model,
26
+ tools=[web_search],
27
+ managed_agents=[],
28
+ # class='CodeAgent',
29
+ max_steps=20,
30
+ verbosity_level=1,
31
+ grammar=None,
32
+ planning_interval=None,
33
+ name=None,
34
+ description=None,
35
+ executor_type='local',
36
+ executor_kwargs={},
37
+ max_print_outputs_length=None,
38
+ prompt_templates=prompt_templates
39
+ )
40
+ if __name__ == "__main__":
41
+ GradioUI(agent).launch()