SherwinDesouza commited on
Commit
6454750
·
verified ·
1 Parent(s): fb66478

Update tools/transcription_tool.py

Browse files
Files changed (1) hide show
  1. tools/transcription_tool.py +1 -1
tools/transcription_tool.py CHANGED
@@ -4,7 +4,7 @@ from smolagents.tools import Tool
4
  class YoutubeTranscriptions(Tool):
5
  name = "Transcription Generator"
6
  description = "Generates transcription for a youtube video and summarizes them. "
7
- inputs = {'query': {'type': 'string', 'description': ' URL of the video to transcribe'}}
8
  output_type = "string"
9
  def forward(self, URL: str) -> str:
10
  transcription = YouTubeTranscriptApi.get_transcript(URL)
 
4
  class YoutubeTranscriptions(Tool):
5
  name = "Transcription Generator"
6
  description = "Generates transcription for a youtube video and summarizes them. "
7
+ inputs = {'URL': {'type': 'string', 'description': ' URL of the video to transcribe'}}
8
  output_type = "string"
9
  def forward(self, URL: str) -> str:
10
  transcription = YouTubeTranscriptApi.get_transcript(URL)