Update app.py

app.py

@@ -1,12 +1,16 @@
 import gradio as gr
 import whisper
 from pytube import YouTube
+from whisper import dowload_youtube_video, transcribe_audio
 
 loaded_model = whisper.load_model("medium")
 current_size = 'medium'
 def inference(link):
   yt = YouTube(link)
-  path = yt.streams.get_audio_only().download(mp3=True)
+  #path = yt.streams.get_audio_only().download(mp3=True)
+
+  path = dowload_youtube_video(link)
+    
   options = whisper.DecodingOptions(language= 'Spanish', without_timestamps=True)
   results = loaded_model.transcribe(path)
   return results['text']