pbanavara commited on
Commit
44a9754
·
verified ·
1 Parent(s): c767b65

Update app.py

Browse files

Changed Qwen model to 7B-Instruct
Added web page and search tools

Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -4,6 +4,8 @@ import requests
4
  import pytz
5
  import yaml
6
  from tools.final_answer import FinalAnswerTool
 
 
7
 
8
  from Gradio_UI import GradioUI
9
 
@@ -44,10 +46,14 @@ def get_dad_joke() -> str:
44
  return f"Error fetching joke"
45
 
46
  final_answer = FinalAnswerTool()
 
 
 
 
47
  model = HfApiModel(
48
  max_tokens=2096,
49
  temperature=0.5,
50
- model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
51
  custom_role_conversions=None,
52
  )
53
 
@@ -60,7 +66,7 @@ with open("prompts.yaml", 'r') as stream:
60
 
61
  agent = CodeAgent(
62
  model=model,
63
- tools=[final_answer, get_current_time_in_timezone, get_dad_joke],
64
  max_steps=6,
65
  verbosity_level=1,
66
  grammar=None,
 
4
  import pytz
5
  import yaml
6
  from tools.final_answer import FinalAnswerTool
7
+ from tools.visit_webpage import VisitWebpageTool
8
+ from tools.web_search import DuckDuckGoSearchTool
9
 
10
  from Gradio_UI import GradioUI
11
 
 
46
  return f"Error fetching joke"
47
 
48
  final_answer = FinalAnswerTool()
49
+ webpage_tool = VisitWebpageTool()
50
+ search_tool = DuckDuckGoSearchTool()
51
+
52
+ visit_webpage =
53
  model = HfApiModel(
54
  max_tokens=2096,
55
  temperature=0.5,
56
+ model_id='Qwen/Qwen2.5-Coder-7B-Instruct',
57
  custom_role_conversions=None,
58
  )
59
 
 
66
 
67
  agent = CodeAgent(
68
  model=model,
69
+ tools=[final_answer, get_current_time_in_timezone, get_dad_joke, webpage_tool, search_tool],
70
  max_steps=6,
71
  verbosity_level=1,
72
  grammar=None,