Update app.py

app.py

@@ -167,10 +167,11 @@ import torch
 
 import gradio as gr
 import yt_dlp as youtube_dl
-import pytube as pt
+from pytubefix import YouTube
+from pytubefix.cli import on_progress
+
 from transformers import pipeline
 from transformers.pipelines.audio_utils import ffmpeg_read
-import pytube
 
 import tempfile
 import os
@@ -190,7 +191,7 @@ pipe = pipeline(
 )
 
 
-@spaces.GPU
+# @spaces.GPU
 def transcribe(inputs, task="transcribe"):
     if inputs is None:
         raise gr.Error("No audio file submitted! Please upload or record an audio file before submitting your request.")
@@ -264,15 +265,20 @@ def yt_transcribe(yt_url, task="transcribe", progress=gr.Progress(), max_filesiz
     progress(0, desc="Loading audio file...")
     html_embed_str = _return_yt_html_embed(yt_url)
     try:
-        yt = pytube.YouTube(yt_url)
-        stream = yt.streams.filter(only_audio=True)[0]
+        # yt = pytube.YouTube(yt_url)
+        # stream = yt.streams.filter(only_audio=True)[0]
+        yt = YouTube(yt_url, on_progress_callback = on_progress)
+         
+        stream = yt.streams.get_audio_only()
+        
     except:
         raise gr.Error("An error occurred while loading the YouTube video. Please try again.")
 
     if stream.filesize_mb > max_filesize:
         raise gr.Error(f"Maximum YouTube file size is {max_filesize}MB, got {stream.filesize_mb:.2f}MB.")
 
-    stream.download(filename="audio.mp3")
+    # stream.download(filename="audio.mp3")
+    stream.download(filename="audio.mp3", mp3=True)
 
     with open("audio.mp3", "rb") as f:
         inputs = f.read()