add youtube_dl

app.py

@@ -5,7 +5,7 @@ from PIL import Image
 import pafy
 import gradio as gra
 from transformers import BlipProcessor, BlipForConditionalGeneration
-import subprocess
+import youtube_dl
 
 model_id = "Salesforce/blip-image-captioning-base"
 
@@ -49,8 +49,18 @@ def user_greeting(name):
     return "Hi! " + name + " Welcome to your first Gradio application!😎"    
     
 # video_url = 'https://www.youtube.com/watch?v=orbkg5JH9C8'
-# subprocess.call(['youtube-dl', '-f', 'best', '-o', 'video.mp4', video_url]
-app = gra.Interface(fn=user_greeting, inputs='text', outputs="text")
+# URL of the YouTube video
+video_url = 'https://www.youtube.com/watch?v=VIDEO_ID'
+
+# Download the video using youtube_dl and extract the frames using OpenCV
+ydl_opts = {
+    'format': 'best',
+    'outtmpl': 'video.mp4'
+}
+with youtube_dl.YoutubeDL(ydl_opts) as ydl:
+    ydl.download([video_url])
+    
+app = gra.Interface(fn=caption_youtube, inputs='text', outputs="text")
 app.launch()