Sandiago21 commited on
Commit
42cc20f
·
verified ·
1 Parent(s): d58d490

Update app.py to enable model creation

Browse files
Files changed (1) hide show
  1. app.py +18 -18
app.py CHANGED
@@ -27,28 +27,28 @@ class BasicAgent:
27
  print(f"Agent returning fixed answer: {fixed_answer}")
28
 
29
 
30
- # model = HfApiModel(
31
- # max_tokens=2096,
32
- # temperature=0.0,
33
- # model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
34
- # custom_role_conversions=None,
35
- # )
36
 
37
  with open("prompts.yaml", 'r') as stream:
38
  prompt_templates = yaml.safe_load(stream)
39
 
40
- # agent = CodeAgent(
41
- # model=model,
42
- # # tools=[DuckDuckGoSearchTool()],
43
- # tools=[],
44
- # max_steps=5,
45
- # verbosity_level=1,
46
- # grammar=None,
47
- # planning_interval=None,
48
- # name=None,
49
- # description=None,
50
- # prompt_templates=prompt_templates
51
- # )
52
 
53
  # agent_answer = agent.run(question)
54
 
 
27
  print(f"Agent returning fixed answer: {fixed_answer}")
28
 
29
 
30
+ model = HfApiModel(
31
+ max_tokens=2096,
32
+ temperature=0.0,
33
+ model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded
34
+ custom_role_conversions=None,
35
+ )
36
 
37
  with open("prompts.yaml", 'r') as stream:
38
  prompt_templates = yaml.safe_load(stream)
39
 
40
+ agent = CodeAgent(
41
+ model=model,
42
+ # tools=[DuckDuckGoSearchTool()],
43
+ tools=[],
44
+ max_steps=5,
45
+ verbosity_level=1,
46
+ grammar=None,
47
+ planning_interval=None,
48
+ name=None,
49
+ description=None,
50
+ prompt_templates=prompt_templates
51
+ )
52
 
53
  # agent_answer = agent.run(question)
54