Update app.py

app.py

@@ -1,29 +1,29 @@
 import whisper
 from pytube import YouTube
-#from transformers import pipeline
+from transformers import pipeline
 import gradio as gr
 import os
 import re
 
 model = whisper.load_model("base")
-#summarizer = pipeline("summarization") 
+summarizer = pipeline("summarization") 
 
-#def get_audio(url):
-    #try:
-    #yt = YouTube(url)
-    #if yt.length < 5400:
-    #video = yt.streams.filter(only_audio=True).first()
-    #out_file=video.download(output_path=".")
-    #base, ext = os.path.splitext(out_file)
-    #new_file = base+'.mp3'
-    #os.rename(out_file, new_file)
-    #a = new_file
-    #return a
-    #else:
-        #raise gr.Error("Videos for transcription on this space are limited to 1.5 hours. Sorry about this limit but some joker thought they could stop this tool from working by transcribing many extremely long videos.")
-        #return ""
-    #finally:
-        #raise gr.Error("Exception: There was a problem getting the video or audio of the URL provided.")
+def get_audio(url):
+    try:
+    yt = YouTube(url)
+    if yt.length < 5400:
+    video = yt.streams.filter(only_audio=True).first()
+    out_file=video.download(output_path=".")
+    base, ext = os.path.splitext(out_file)
+    new_file = base+'.mp3'
+    os.rename(out_file, new_file)
+    a = new_file
+    return a
+    else:
+        raise gr.Error("Videos for transcription on this space are limited to 1.5 hours. Sorry about this limit but some joker thought they could stop this tool from working by transcribing many extremely long videos.")
+        return ""
+    finally:
+        raise gr.Error("Exception: There was a problem getting the video or audio of the URL provided.")
 
 def get_text(url):
     #try: