Create download_aim.py

download_aim.py

@@ -0,0 +1,14 @@
+from huggingface_hub import snapshot_download
+import os
+
+with open('/run/secrets/HF_TOKEN', 'r') as f:
+    token = f.read().strip()
+
+repo_id = os.environ['HF_REPO_ID']
+
+snapshot_download(
+    repo_id=repo_id,
+    local_dir='.',
+    allow_patterns='.aim/**',
+    token=token
+)