mgerlitz commited on
Commit
b80a698
·
verified ·
1 Parent(s): f60987f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -5
app.py CHANGED
@@ -3,7 +3,7 @@ import gradio as gr
3
  import requests
4
  import inspect
5
  import pandas as pd
6
- from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, TransformersModel, VisitWebpageTool, load_tool, tool
7
  import yaml
8
  from tools.final_answer import FinalAnswerTool
9
 
@@ -44,10 +44,18 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
44
  # 1. Instantiate Agent ( modify this part to create your agent)
45
  try:
46
 
47
- model_id = "meta-llama/Llama-3.2-3B-Instruct"
48
-
49
- model = TransformersModel(model_id=model_id)
50
- agent = CodeAgent(tools=[], model=model, add_base_tools=True)
 
 
 
 
 
 
 
 
51
 
52
  #agent = BasicAgent()
53
  except Exception as e:
 
3
  import requests
4
  import inspect
5
  import pandas as pd
6
+ from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, VisitWebpageTool, load_tool, tool
7
  import yaml
8
  from tools.final_answer import FinalAnswerTool
9
 
 
44
  # 1. Instantiate Agent ( modify this part to create your agent)
45
  try:
46
 
47
+ agent = CodeAgent(
48
+ tools=[
49
+ DuckDuckGoSearchTool(),
50
+ VisitWebpageTool(),
51
+ suggest_menu,
52
+ catering_service_tool,
53
+ SuperheroPartyThemeTool()
54
+ ],
55
+ model=HfApiModel(),
56
+ max_steps=10,
57
+ verbosity_level=2
58
+ )
59
 
60
  #agent = BasicAgent()
61
  except Exception as e: