Update app.py

app.py

@@ -4,6 +4,17 @@ import gradio as gr
 
 pipe = pipeline(model="Yilin98/whisper-small-hi")  # change to "your-username/the-name-you-picked"
 
+def get_audio(url):
+  yt = YouTube(url)
+  stream = yt.streams.filter(only_audio=True).first()
+  out_file=stream.download(output_path=".")
+  base, ext = os.path.splitext(out_file)
+  new_file = base+'.mp3'
+  os.rename(out_file, new_file)
+  audio = new_file
+  return audio
+
+
 def transcribe(audio=None, file=None, youtube=None):
     if (audio is None) and (file is None) and (youtube is None):
         return "No audio provided!"
@@ -12,10 +23,7 @@ def transcribe(audio=None, file=None, youtube=None):
     elif file is not None:
         input=file
     elif youtube is not None:
-        yt=YouTube(youtube)
-        stream=yt.streams.filter(only_audio=True)[0]
-        stream.download(filename="audio.mp3")
-        input=audio
+        input=get_audio(youtube)
     text = pipe(input)["text"]
     return text