| ''' |
| import os |
| from huggingface_hub import hf_hub_download |
| from huggingface_hub import login |
| |
| login(token=os.getenv("HF_TOKEN")) |
| |
| model_paths = { |
| "layer1cnn_aanan.pth": hf_hub_download(repo_id="f16sam/awss-models", filename="layer1cnn_aanan.pth"), |
| "layer2bio_00": hf_hub_download(repo_id="f16sam/awss-models", filename="layer2bio_00"), |
| "layer2bio_01": hf_hub_download(repo_id="f16sam/awss-models", filename="layer2bio_01"), |
| "layer2non_00": hf_hub_download(repo_id="f16sam/awss-models", filename="layer2non_00"), |
| "layer2non_01": hf_hub_download(repo_id="f16sam/awss-models", filename="layer2non_01"), |
| "layer3_cnn.keras": hf_hub_download(repo_id="f16sam/awss-models", filename="layer3_cnn.keras") |
| } |
| |
| |
| # Cache Folder |
| custom_cache = "/tmp/.cache" |
| os.makedirs(custom_cache, exist_ok=True) |
| os.environ["XDG_CACHE_HOME"] = custom_cache |
| os.environ["GDOWN_CACHE_DIR"] = custom_cache |
| import gdown |
| |
| # Destination folder |
| MODEL_DIR = os.path.join(os.path.dirname(__file__), ".") |
| files_to_download = { |
| "layer1cnn_aanan.pth": "1R6Up_9vyd27hdRIyGXQgi86pn42tl-bh", |
| "layer2bio_cnn.keras": "1mewJKVzhmOl_l-sVupK3OasyX_56OxjG", |
| "layer2non_cnn.keras": "1IM6Y4ZduHE4JeDgJig362n-Y1tz6YKmB", |
| "layer3_cnn.keras": "16DjttPx1f9sqWlodO5KqPyZBkpYVtVa4" |
| } |
| |
| def download_all_models(): |
| for filename, file_id in files_to_download.items(): |
| dest_path = os.path.join(MODEL_DIR, filename) |
| if not os.path.exists(dest_path): |
| print(f"Downloading {filename}...") |
| gdown.download(f"https://drive.google.com/uc?id={file_id}", dest_path, quiet=False) |
| else: |
| print(f"{filename} already exists, skipping.") |
| |
| if __name__ == "__main__": |
| download_all_models() |
| ''' |
