Update models.py

models.py

@@ -6,16 +6,17 @@ import os
 from safetensors.torch import load_file  # Safetensors loading method
 
 # Set a custom cache directory for gdown
-os.makedirs("cache", exist_ok=True)
-os.environ["GDOWN_CACHE"] = os.path.abspath("cache")
+gdown_cache_dir = os.path.join(os.getcwd(), "cache")
+os.makedirs(gdown_cache_dir, exist_ok=True)
+os.environ["GDOWN_CACHE"] = gdown_cache_dir  # Explicitly set GDOWN_CACHE
 
 def download_model_from_drive(file_id, destination_path):
     """Download the model from Google Drive using gdown."""
     # Construct the Google Drive download URL
     url = f"https://drive.google.com/uc?id={file_id}"
-    # Extract the directory name from the destination path
+    # Ensure the destination directory exists
     directory = os.path.dirname(destination_path)
-    if directory:  # Ensure the directory exists only if it's valid
+    if directory:
         os.makedirs(directory, exist_ok=True)
     # Download the file
     gdown.download(url, destination_path, quiet=False)