download url method

requirements.txt

@@ -20,4 +20,5 @@ gradio==4.1.2
 ffmpy
 flask_ngrok2
 flask_ngrok
-opencv-python
+opencv-python
+requests

run.py

@@ -1,11 +1,19 @@
 import os
 import sys
-from urllib.request import urlretrieve
+import requests
 
 from main import call_wav2lip, call_gfpgan, merge
 
 root_dir = '/content/jobs'
-os.makedirs(root_dir,exist_ok=True)
+os.makedirs(root_dir, exist_ok=True)
+
+
+def download_file(url, local_filename):
+    response = requests.get(url, stream=True)
+    with open(local_filename, 'wb') as file:
+        for chunk in response.iter_content(chunk_size=8192):
+            if chunk:
+                file.write(chunk)
 
 
 def main(job_id, video_url, audio_url):
@@ -15,14 +23,14 @@ def main(job_id, video_url, audio_url):
     if video_url.startswith('http'):
         video_file = os.path.basename(video_url)
         video_path = os.path.join(job_path, video_file)
-        urlretrieve(video_url, video_path)
+        download_file(video_url, video_path)
     else:
         video_path = video_url
 
     if audio_url.startswith('http'):
         audio_file = os.path.basename(audio_url)
         audio_path = os.path.join(job_path, audio_file)
-        urlretrieve(audio_url, audio_path)
+        download_file(audio_url, audio_path)
     else:
         audio_path = audio_url