Update summarize.py

summarize.py

@@ -4,8 +4,17 @@ import sys
 from youtube_transcript_api import YouTubeTranscriptApi
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
+
+def add_watch_word(link):
+    if 'watch?' in link or 'embed/' in link or 'v=' in link:
+        return link
+    else:
+        split_link = link.split('/')
+        last_part = split_link[-1]
+        return '/'.join(split_link[:-1]) + '/watch?v=' + last_part
+        
 def Summarizer(link, model):
-  
+  link=add_watch_word(link)
   video_id = link.split("=")[1]
 
   try: