update test

app.py

@@ -2,7 +2,7 @@ import time
 import os
 
 import torch
-
+import yt_dlp
 import gradio as gr
 import pytube as pt
 import spaces
@@ -66,30 +66,23 @@ def _return_yt_html_embed(yt_url):
 
 
 def yt_transcribe(yt_url, return_timestamps=False):
-    try:
-        yt = pt.YouTube(yt_url)
-    except Exception as e:
-        return f"Error fetching YouTube video: {str(e)}"
-
-    html_embed_str = _return_yt_html_embed(yt_url)
-    
-    audio_streams = yt.streams.filter(only_audio=True)
-    if not audio_streams:
-        return "No audio streams available for this video."
-    
-    stream = audio_streams[0]
+    ydl_opts = {
+        'format': 'bestaudio/best',
+        'outtmpl': 'audio.mp3',
+        'noplaylist': True
+    }
 
     try:
-        stream.download(filename="audio.mp3")
+        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            ydl.download([yt_url])
     except Exception as e:
         return f"Error downloading audio: {str(e)}"
-    
+
     if not os.path.exists("audio.mp3"):
         return "Downloaded audio file not found."
 
     text = transcribe("audio.mp3", return_timestamps=return_timestamps)
-    
-    return html_embed_str, text
+    return _return_yt_html_embed(yt_url), text
 
 
 demo = gr.Blocks()