fix: cookies didn't work correctly

app.py

@@ -12,6 +12,7 @@ from pathlib import Path
 import spaces
 from pydub.exceptions import CouldntEncodeError
 from transformers import pipeline
+import requests
 
 # Initialize text generation model
 model = pipeline('text-generation', model='EleutherAI/gpt-neo-125M')
@@ -21,6 +22,22 @@ OUTPUT_FOLDER = "separation_results/"
 INPUT_FOLDER = "input"
 download_path = ""
 
+# URL for the cookies.txt file in the Hugging Face repository
+cookies_url = "https://huggingface.co/spaces/Awell00/music_drums_separation/raw/main/cookies.txt"
+
+def download_cookies():
+    try:
+        response = requests.get(cookies_url)
+        response.raise_for_status()  # Check for HTTP errors
+        
+        # Write content to cookies.txt file in the Docker container
+        with open("cookies.txt", "w") as file:
+            file.write(response.text)
+        print("cookies.txt downloaded successfully.")
+        
+    except requests.exceptions.RequestException as e:
+        print(f"Error downloading cookies.txt: {e}")
+
 class MyLogger:
     def debug(self, msg):
         # For compatibility with youtube-dl, both debug and info are passed into debug
@@ -145,6 +162,8 @@ def download_youtube_audio(youtube_url: str, output_dir: str = './download', del
     if not os.path.exists(output_dir):
         os.makedirs(output_dir)
 
+    download_cookies()
+
     title = get_video_title(youtube_url)
 
     audio_file = os.path.join(output_dir, title)