Update application

app.py

@@ -115,6 +115,7 @@ with gr.Blocks(css="style.css") as demo:
 
  **Note**: 
  - It will take a bit of time to complete (~40% of the original video length), so stay tuned!
+ - If the YouTube downloader return errors, you can try again later or download video to your computer and then upload it to the app
  - Remember to press Enter if you are using an external URL
  """,
  elem_id="container",

config.py

@@ -2,7 +2,7 @@ import imagehash
 
 # -------------- Initializations ---------------------
 
-DOWNLOAD_DIR = "/tmp"
+DOWNLOAD_DIR = "downloads"
 
 FRAME_BUFFER_HISTORY = 15 # Length of the frame buffer history to model background.
 DEC_THRESH = (

download_video.py

@@ -1,4 +1,5 @@
 import mimetypes
+import re
 import tempfile
 import requests
 import os
@@ -57,6 +58,7 @@ def download_video_from_youtube(url, output_dir=DOWNLOAD_DIR):
 def download_video(url, output_dir=DOWNLOAD_DIR):
  parsed_url = urlparse(url)
  domain = parsed_url.netloc.lower()
+ domain = re.sub(r"\.", "", domain) # Match for both youtube and youtu.be
 
  print("---" * 5, "Downloading video file", "---" * 5)