switch-to-yt-dlp

app.py

@@ -1,6 +1,6 @@
 import torch
 import clip
-import cv2, youtube_dl
+import cv2, yt_dlp
 from PIL import Image,ImageDraw, ImageFont
 import wget
 import os
@@ -21,7 +21,7 @@ model, preprocess = clip.load("ViT-B/32")
 def select_video_format(url, ydl_opts={}, format_note='240p', ext='mp4', max_size = 500000000):
     defaults = ['480p', '360p','240p','144p']
     ydl_opts = ydl_opts
-    ydl = youtube_dl.YoutubeDL(ydl_opts)
+    ydl = yt_dlp.YoutubeDL(ydl_opts)
     info_dict = ydl.extract_info(url, download=False)
     formats = info_dict.get('formats', None)
     # filter out formats we can't process
@@ -62,7 +62,7 @@ def download_video(url):
         'format':format_id,
         'outtmpl': "videos/%(id)s.%(ext)s"}
 
-      with youtube_dl.YoutubeDL(ydl_opts) as ydl:
+      with yt_dlp.YoutubeDL(ydl_opts) as ydl:
         try:
           ydl.cache.remove()
           meta = ydl.extract_info(url)
@@ -71,7 +71,7 @@ def download_video(url):
           thumb_location = f"videos/{meta['id']}.{thumb_url.split('.')[-1]}"
           wget.download(thumb_url, out=thumb_location)
           save_location = 'videos/' + meta['id'] + '.' + meta['ext']
-        except youtube_dl.DownloadError as error:
+        except yt_dlp.DownloadError as error:
           print(f'error with download_video function: {error}')
           save_location = None
           thumb_location = None

requirements.txt

@@ -1,5 +1,5 @@
 git+https://github.com/openai/CLIP.git
 torch
-youtube_dl
+yt-dlp
 opencv-python-headless
 wget