Preetesh/VideoSummaryfromYouTubeVideo

1import traceback2import sys3 4from youtube_transcript_api import YouTubeTranscriptApi5from transformers import AutoTokenizer, AutoModelForSeq2SeqLM6 7def Summarizer(link, model):8  9  video_id = link.split("=")[1]10 11  try:12    transcript = YouTubeTranscriptApi.get_transcript(video_id)13    FinalTranscript = ' '.join([i['text'] for i in transcript])14    15    if model == "Pegasus":16      checkpoint = "google/pegasus-large"17    elif model == "mT5":18      checkpoint = "csebuetnlp/mT5_multilingual_XLSum"19    elif model == "BART":20      checkpoint = "sshleifer/distilbart-cnn-12-6"21      22    tokenizer = AutoTokenizer.from_pretrained(checkpoint)23    model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint)24 25 26    inputs = tokenizer(FinalTranscript, 27                    max_length=1024, 28                    truncation=True,29                    return_tensors="pt")30    31    summary_ids = model.generate(inputs["input_ids"])32    summary = tokenizer.batch_decode(summary_ids, 33                                  skip_special_tokens=True, 34                                  clean_up_tokenization_spaces=False)35    36 37    return summary[0]38    39 40  except Exception:41    print(traceback.format_exc())42    # or43    print(sys.exc_info()[2])