CindyDelage commited on
Commit
1eb2891
·
verified ·
1 Parent(s): 63b8e47

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -3,7 +3,7 @@ import gradio as gr
3
  import requests
4
  import inspect
5
  import pandas as pd
6
- from tools import Web_research, image_interpreter, Wikipedia_reader, translate_everything
7
  from smolagents import GradioUI, CodeAgent, HfApiModel, PythonInterpreterTool
8
 
9
  # (Keep Constants as is)
@@ -23,8 +23,9 @@ class BasicAgent:
23
  python_code_tool = PythonInterpreterTool()
24
  wiki_tool = Wikipedia_reader()
25
  translator=translate_everything()
 
26
  alfred=CodeAgent(
27
- tools=[translator,wiki_tool,web_search, image_tool,python_code_tool],
28
  model=HfApiModel(token=os.environ.get("HF_TOKEN")),
29
  add_base_tools=True,
30
  additional_authorized_imports=['urllib','chess','requests', 'bs4'],
 
3
  import requests
4
  import inspect
5
  import pandas as pd
6
+ from tools import Web_research, image_interpreter, Wikipedia_reader, translate_everything, Find_wikipedia_URL
7
  from smolagents import GradioUI, CodeAgent, HfApiModel, PythonInterpreterTool
8
 
9
  # (Keep Constants as is)
 
23
  python_code_tool = PythonInterpreterTool()
24
  wiki_tool = Wikipedia_reader()
25
  translator=translate_everything()
26
+ wiki_url_tool=Find_wikipedia_URL()
27
  alfred=CodeAgent(
28
+ tools=[translator,wiki_tool,wiki_url_tool,web_search, image_tool,python_code_tool],
29
  model=HfApiModel(token=os.environ.get("HF_TOKEN")),
30
  add_base_tools=True,
31
  additional_authorized_imports=['urllib','chess','requests', 'bs4'],