WingNeville commited on
Commit
a8168d6
·
verified ·
1 Parent(s): 805622e

Upload agent

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