jamesbright commited on
Commit
8135ce6
·
verified ·
1 Parent(s): e639f71

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +21 -7
app.py CHANGED
@@ -1,11 +1,14 @@
1
  import gradio as gr
2
  import json
3
- from smolagents import CodeAgent, tool, Tool
4
- from smolagents.models
5
- import HfApiModel
6
 
7
- # Initialize the model
8
- model = HfApiModel(model_id="Qwen/Qwen2.5-Coder-32B-Instruct")
 
 
 
 
 
9
 
10
 
11
  class ProjectScopingTool(Tool):
@@ -28,7 +31,8 @@ class ProjectScopingTool(Tool):
28
  self.project_details = {
29
  "type": project_type,
30
  "domain": domain,
31
- "budget_range": {"min": budget_min, "max": budget_max},
 
32
  "timeline": {"months": timeline_months}
33
  }
34
  return json.dumps(self.project_details, indent=2)
@@ -121,7 +125,17 @@ tools = [
121
  meeting_tool.generate_meeting_agenda
122
  ]
123
 
124
- agent = CodeAgent(tools=tools, model=model)
 
 
 
 
 
 
 
 
 
 
125
 
126
  # Define Gradio Interface
127
  with gr.Blocks() as demo:
 
1
  import gradio as gr
2
  import json
3
+ from smolagents import CodeAgent, tool, Tool, FinalAnswerTool, HfApiModel, load_tool
 
 
4
 
5
+ final_answer = FinalAnswerTool()
6
+ model = HfApiModel(
7
+ max_tokens=2096,
8
+ temperature=0.5,
9
+ model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
10
+ custom_role_conversions=None,
11
+ )
12
 
13
 
14
  class ProjectScopingTool(Tool):
 
31
  self.project_details = {
32
  "type": project_type,
33
  "domain": domain,
34
+ "budget_range": {"min"
35
+ : budget_min, "max": budget_max},
36
  "timeline": {"months": timeline_months}
37
  }
38
  return json.dumps(self.project_details, indent=2)
 
125
  meeting_tool.generate_meeting_agenda
126
  ]
127
 
128
+ agent = CodeAgent(
129
+ model=model,
130
+ tools=[final_answer, tools], # add your tools here (don't remove final_answer)
131
+ max_steps=6,
132
+ verbosity_level=1,
133
+ grammar=None,
134
+ planning_interval=None,
135
+ name=None,
136
+ description=None,
137
+ prompt_templates=prompt_templates
138
+ )
139
 
140
  # Define Gradio Interface
141
  with gr.Blocks() as demo: