Update app.py

app.py

@@ -1,20 +1,6 @@
 import gradio as gr
 import whisper
-#from pytube import YouTube
-import os
-from googleapiclient.discovery import build
-
-api_key = os.environ['api_key']
-youtube_api = build('youtube', 'v3', developerKey=api_key)
-
-#Get title and thumbnail
-def get_vid_details(youtube_link):
-    youtube_id = get_video_id(youtube_link)
-    return response['items'][0]['snippet']['thumbnails']['high']['url'], youtube_id['items'][0]['snippet']['title']
-
-def get_video_id(url):
-    parsed_url = urlparse(url)
-    return parse_qs(parsed_url.query)['v'][0]
+from pytube import YouTube
 
 loaded_model = whisper.load_model("medium")
 current_size = 'medium'
@@ -32,9 +18,8 @@ def change_model(size):
   current_size = size
 
 def populate_metadata(link):
-  #yt = YouTube(link)
-  #return yt.thumbnail_url, yt.title
-    return thumbnnail(link), caption(link)
+  yt = YouTube(link)
+  return yt.thumbnail_url, yt.title
 
 title=""
 description=""
@@ -70,8 +55,7 @@ with block:
           
           # Events
           btn.click(inference, inputs=[link], outputs=[text])
-          #link.change(populate_metadata, inputs=[link], outputs=[img, title])
-          link.change(get_vid_details, inputs=[link], outputs=[img, title])
+          link.change(populate_metadata, inputs=[link], outputs=[img, title])
           sz.change(change_model, inputs=[sz], outputs=[])
 
 block.launch(debug=True)