Update app.py

app.py

@@ -1,18 +1,18 @@
 from transformers import pipeline
 import gradio as gr
+from pytube import YouTube
 
 pipe = pipeline(model="Akseluhr/whisper-small-sv-SE-auhr-v2")
 
 def get_audio(url):
-  print(url)
   yt = YouTube(url)
   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
+  audio_file = new_file
+  return audio_file
 
 def get_text(url):
   result = model.transcribe(get_audio(url))
@@ -24,7 +24,6 @@ def transcribe(rec=None, file=None, url=None):
     elif file is not None:
         audio = file
     elif url is not None:
-        print(url)
         audio = get_audio(url)
     else:
         return "Provide a recording or a file."