Update app.py

app.py

@@ -1,15 +1,20 @@
 import cv2 # opencv2 package for python.
-import pafy # pafy allows us to read videos from youtube.
+from pytube import YouTube
 from torch import hub # Hub contains other models like FasterRCNN
 
 URL = "https://www.youtube.com/watch?v=dQw4w9WgXcQ" #URL to parse
-play = pafy.new(_URL).streams[-1] #'-1' means read the lowest quality of video.
+#play = pafy.new(_URL).streams[-1] #'-1' means read the lowest quality of video.
 
-assert play is not None # we want to make sure their is a input to read.
-stream = cv2.VideoCapture(play.url) #create a opencv video stream.
+#assert play is not None # we want to make sure their is a input to read.
+#stream = cv2.VideoCapture(play.url) #create a opencv video stream.
 #stream = cv2.VideoCapture(0) # 0 means read from local camera.
 #camera_ip = "rtsp://username:password@IP/port"
 #stream = cv2.VideoCapture(camera_ip)
+def load():
+    yt = YouTube(URL)
+    vid_cap = yt.streams.filter(progressive=True, file_extension='mp4').order_by('resolution').desc().first().download(filename="tmp.mp4")
+    stream = cv2.VideoCapture(vid)
+    return vid_cap
 
 model = torch.hub.load('ultralytics/yolov5','yolov5s',pretrained=True)