SherwinDesouza commited on
Commit
57db83d
·
verified ·
1 Parent(s): a0b3e68

Update tools/transcription_tool.py

Browse files
Files changed (1) hide show
  1. tools/transcription_tool.py +2 -0
tools/transcription_tool.py CHANGED
@@ -5,6 +5,8 @@ class YoutubeTranscriptions(Tool):
5
  def __init__(self):
6
  self.name = "YoutubeTranscriptions"
7
  self.description = "Generates transcription for a YouTube video. If transcription cannot be generated, return 'No transcriptions available.'"
 
 
8
 
9
  def forward(self, URL: str) -> str:
10
  # Extract video ID from URL
 
5
  def __init__(self):
6
  self.name = "YoutubeTranscriptions"
7
  self.description = "Generates transcription for a YouTube video. If transcription cannot be generated, return 'No transcriptions available.'"
8
+ self.inputs = {'URL': {'type': 'string', 'description': 'URL of the video to transcribe'}}
9
+ self.output_type = "string"
10
 
11
  def forward(self, URL: str) -> str:
12
  # Extract video ID from URL