sachin261 commited on
Commit
9f44c87
·
verified ·
1 Parent(s): dad57de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
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,7 @@ 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 +32,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
+ agent_class='CodeAgent', # ✅ Renamed from "class"
24
  max_steps=20,
25
  verbosity_level=1,
26
  grammar=None,
 
32
  max_print_outputs_length=None,
33
  prompt_templates=prompt_templates
34
  )
35
+
36
  if __name__ == "__main__":
37
  GradioUI(agent).launch()