WingNeville commited on
Commit
b07b6d2
·
verified ·
1 Parent(s): 1e5c7d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -6
app.py CHANGED
@@ -7,15 +7,12 @@ CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
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,7 +20,6 @@ agent = CodeAgent(
23
  model=model,
24
  tools=[],
25
  managed_agents=[],
26
- class='CodeAgent',
27
  max_steps=20,
28
  verbosity_level=1,
29
  grammar=None,
@@ -35,5 +31,6 @@ agent = CodeAgent(
35
  max_print_outputs_length=None,
36
  prompt_templates=prompt_templates
37
  )
 
38
  if __name__ == "__main__":
39
- GradioUI(agent).launch()
 
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
  model=model,
21
  tools=[],
22
  managed_agents=[],
 
23
  max_steps=20,
24
  verbosity_level=1,
25
  grammar=None,
 
31
  max_print_outputs_length=None,
32
  prompt_templates=prompt_templates
33
  )
34
+
35
  if __name__ == "__main__":
36
+ GradioUI(agent).launch()