Update app.py

app.py

@@ -1,32 +1,21 @@
 import gradio as gr
 import whisper
-from pytube import YouTube
-#from pytubefix import YouTube
-#from pytubefix.cli import on_progress
+#from pytube import YouTube
+from pytubefix import YouTube
+from pytubefix.cli import on_progress
 
 loaded_model = whisper.load_model("medium")
 current_size = 'medium'
 
-
-from pytube. innertube import _default_clients
-
-_default_clients[ "ANDROID"][ "context"]["client"]["clientVersion"] = "19.08.35"
-_default_clients["IOS"]["context"]["client"]["clientVersion"] = "19.08.35"
-_default_clients[ "ANDROID_EMBED"][ "context"][ "client"]["clientVersion"] = "19.08.35"
-_default_clients[ "IOS_EMBED"][ "context"]["client"]["clientVersion"] = "19.08.35"
-_default_clients["IOS_MUSIC"][ "context"]["client"]["clientVersion"] = "6.41"
-_default_clients[ "ANDROID_MUSIC"] = _default_clients[ "ANDROID_CREATOR" ]
-
-
-
-
 def inference(link):
-  yt = YouTube(link)
+  #yt = YouTube(link)
   #yt = YouTube(link, on_progress_callback=on_progress, use_po_token=True)
-  global audio_stream
-  audio_stream = yt.streams.filter(only_audio=True, file_extension='mp4').first()
-  path = audio_stream.download()
+  yt = YouTube(link, on_progress_callback=on_progress)
+  #global audio_stream
+  #audio_stream = yt.streams.filter(only_audio=True, file_extension='mp4').first()
+  #path = audio_stream.download()
   #path = yt.streams.get_audio_only().download(mp3=True)
+  path = yt.streams.get_audio_only().download()
   options = whisper.DecodingOptions(language= 'Spanish', without_timestamps=True)
   results = loaded_model.transcribe(path)
   return results['text']