RemVdH commited on
Commit
b075a3f
·
verified ·
1 Parent(s): 6253a89

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -1,8 +1,8 @@
1
  from smolagents import CodeAgent, HfApiModel, load_tool, tool
2
  from tools.final_answer import FinalAnswerTool
3
- from tools.youtube_search import YoutubeSearchTool
4
- from tools.play_music import PlaySomeMusicMeastro
5
- from tools.download_music import DownloadSomeMusicMeastro
6
  from typing import List
7
  from Gradio_UI import GradioUI
8
 
@@ -71,7 +71,8 @@ with open("prompts.yaml", 'r') as stream:
71
 
72
  agent = CodeAgent(
73
  model=model,
74
- tools=[final_answer, search_wikipedia_pages, get_wikipedia_page_content, PlaySomeMusicMeastro(), YoutubeSearchTool(), DownloadSomeMusicMeastro(), image_generation_tool],
 
75
  max_steps=12,
76
  verbosity_level=1,
77
  grammar=None,
 
1
  from smolagents import CodeAgent, HfApiModel, load_tool, tool
2
  from tools.final_answer import FinalAnswerTool
3
+ # from tools.youtube_search import YoutubeSearchTool
4
+ # from tools.play_music import PlaySomeMusicMeastro
5
+ # from tools.download_music import DownloadSomeMusicMeastro
6
  from typing import List
7
  from Gradio_UI import GradioUI
8
 
 
71
 
72
  agent = CodeAgent(
73
  model=model,
74
+ # tools=[final_answer, search_wikipedia_pages, get_wikipedia_page_content, PlaySomeMusicMeastro(), YoutubeSearchTool(), DownloadSomeMusicMeastro(), image_generation_tool],
75
+ tools=[final_answer, search_wikipedia_pages, get_wikipedia_page_content, image_generation_tool],
76
  max_steps=12,
77
  verbosity_level=1,
78
  grammar=None,