Update Dockerfile

Dockerfile

@@ -28,8 +28,10 @@ ENTRYPOINT ["/bin/sh", "-c"]
 # RUN --mount=type=secret,id=HF_TOKEN,mode=0444,required=true \
 	# curl test -H 'Authorization: Bearer $(cat /run/secrets/HF_TOKEN)'
 
+COPY download_aim.py .
+
 RUN --mount=type=secret,id=HF_TOKEN,mode=0444,required=true \
-    python3 -c "from huggingface_hub import snapshot_download; import os; with open('/run/secrets/HF_TOKEN', 'r') as f: token = f.read().strip(); snapshot_download(repo_id='LocalResearchGroup/smollm2-135m-20250224_023514', local_dir='.', allow_patterns='.aim/**', token=token)"
+    python3 download_aim.py
 
 # have to run `aim init` in the directory that stores aim data for
 # otherwise `aim up` will prompt for confirmation to create the directory itself.