Sonu313131 commited on
Commit
29a842e
·
verified ·
1 Parent(s): d379226

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -22,14 +22,16 @@ import re
22
 
23
  from smolagents import Tool
24
 
 
 
25
  class YouTubeTranscriptTool(Tool):
26
  name = "youtube_transcript"
27
  description = "Extracts transcript text from a YouTube video URL"
28
  inputs = {"url": str}
29
  outputs = {"transcript": str}
 
30
 
31
  def __call__(self, url: str) -> dict:
32
- # Example logic (replace with real code)
33
  from youtube_transcript_api import YouTubeTranscriptApi
34
  video_id = url.split("v=")[-1].split("&")[0]
35
  transcript_list = YouTubeTranscriptApi.get_transcript(video_id)
 
22
 
23
  from smolagents import Tool
24
 
25
+ from smolagents import Tool
26
+
27
  class YouTubeTranscriptTool(Tool):
28
  name = "youtube_transcript"
29
  description = "Extracts transcript text from a YouTube video URL"
30
  inputs = {"url": str}
31
  outputs = {"transcript": str}
32
+ output_type = dict # ✅ This line is required!
33
 
34
  def __call__(self, url: str) -> dict:
 
35
  from youtube_transcript_api import YouTubeTranscriptApi
36
  video_id = url.split("v=")[-1].split("&")[0]
37
  transcript_list = YouTubeTranscriptApi.get_transcript(video_id)