Delete summarize.py

summarize.py

@@ -1,43 +0,0 @@
-import traceback
-import sys
-
-from youtube_transcript_api import YouTubeTranscriptApi
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
-
-def Summarizer(link, model):
-  
-  video_id = link.split("=")[1]
-
-  try:
-    transcript = YouTubeTranscriptApi.get_transcript(video_id)
-    FinalTranscript = ' '.join([i['text'] for i in transcript])
-    
-    if model == "Pegasus":
-      checkpoint = "google/pegasus-large"
-    elif model == "mT5":
-      checkpoint = "csebuetnlp/mT5_multilingual_XLSum"
-    elif model == "BART":
-      checkpoint = "sshleifer/distilbart-cnn-12-6"
-      
-    tokenizer = AutoTokenizer.from_pretrained(checkpoint)
-    model = AutoModelForSeq2SeqLM.from_pretrained(checkpoint)
-
-
-    inputs = tokenizer(FinalTranscript, 
-                    max_length=1024, 
-                    truncation=True,
-                    return_tensors="pt")
-    
-    summary_ids = model.generate(inputs["input_ids"])
-    summary = tokenizer.batch_decode(summary_ids, 
-                                  skip_special_tokens=True, 
-                                  clean_up_tokenization_spaces=False)
-    
-
-    return summary[0]
-    
-
-  except Exception:
-    print(traceback.format_exc())
-    # or
-    print(sys.exc_info()[2])