Hugging Face's logo

Spaces:
Sborole
/
Sborole-Final-Assignment
Sleeping

App Files Community
Sborole commited on
Commit
6391ed8
·
verified ·
1 Parent(s): bacd706

Update tools/YouTubeTool.py

Browse files
Files changed (1) hide show
  1. tools/YouTubeTool.py +2 -2
tools/YouTubeTool.py CHANGED
@@ -2,9 +2,9 @@ from smolagents import Tool
2
  from youtube_transcript_api import YouTubeTranscriptApi
3
 
4
  class YouTubeTool(Tool):
5
- name = "youtube reader"
6
  description= "Fetches the transcript for the youtube video url"
7
- inputs= {"video_url": {"type": "string"}, "description": "URL or ID of the Youtube video"}
8
  output_type = "string"
9
 
10
  def forward(self, video_url: str) -> str:
 
2
  from youtube_transcript_api import YouTubeTranscriptApi
3
 
4
  class YouTubeTool(Tool):
5
+ name = "youtube_reader"
6
  description= "Fetches the transcript for the youtube video url"
7
+ inputs= {"video_url": {"type": "string", "description": "URL or ID of the Youtube video"}}
8
  output_type = "string"
9
 
10
  def forward(self, video_url: str) -> str: