switch-to-yt-dlp-for-faster-dl (#3)

- switch-to-yt-dlp-for-faster-dl (f6e845c6efb89ae69c9360404c976107fa7b3ed4)


Co-authored-by: Ernie Chu <[email protected]>

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 os
 from functools import partial
@@ -20,7 +20,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
@@ -61,12 +61,12 @@ 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)
           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
     except IndexError as err:

requirements.txt

@@ -1,4 +1,4 @@
 git+https://github.com/openai/CLIP.git
 torch
-youtube_dl
-opencv-python-headless
+yt_dlp
+opencv-python-headless