Update app.py

app.py

@@ -66,6 +66,13 @@ saved_cfg.task.label_dir = data_dir
 task = tasks.setup_task(saved_cfg.task)
 generator = task.build_generator(models, gen_cfg)
 
+def get_youtube(video_url):
+    yt = YouTube(video_url)
+    abs_video_path = yt.streams.filter(progressive=True, file_extension='mp4').order_by('resolution').desc().first().download()
+    print("Success download video")
+    print(abs_video_path)
+    return abs_video_path
+    
 def detect_landmark(image, detector, predictor):
     gray = cv2.cvtColor(image, cv2.COLOR_RGB2GRAY)
     face_locations  = detector(gray, 1)