modified links for Angione-Lab

Dockerfile

@@ -30,8 +30,8 @@ RUN mkdir -p ckp && chown user:user ckp
 
 # Download checkpoints from Hugging Face if ckp is empty (for Space deployment)
 # Requires HF_TOKEN secret in Space Settings (https://huggingface.co/settings/tokens)
-# for private model repos. Set HF_MODEL_REPO to your model repo, e.g. kaveh/Shape2Force
-ARG HF_MODEL_REPO=kaveh/Shape2Force
+# for private model repos. Set HF_MODEL_REPO to your model repo, e.g. Angione-Lab/Shape2Force
+ARG HF_MODEL_REPO=Angione-Lab/Shape2Force
 ENV HF_MODEL_REPO=${HF_MODEL_REPO}
 RUN --mount=type=secret,id=HF_TOKEN,mode=0444,required=true \
     HF_TOKEN=$(cat /run/secrets/HF_TOKEN) python download_ckp.py

README.md

@@ -24,7 +24,7 @@ pip install -r requirements.txt
 streamlit run app.py
 ```
 
-Checkpoints are downloaded automatically from the [Shape2Force model repo](https://huggingface.co/kaveh/Shape2Force) when running in Docker. For local use, place `.pth` files in `ckp/`.
+Checkpoints are downloaded automatically from the [Shape2Force model repo](https://huggingface.co/Angione-Lab/Shape2Force) when running in Docker. For local use, place `.pth` files in `ckp/`.
 
 ## Usage
 

download_ckp.py

@@ -11,7 +11,7 @@ if not has_any:
     try:
         from huggingface_hub import hf_hub_download, list_repo_files
 
-        repo = os.environ.get("HF_MODEL_REPO", "kaveh/Shape2Force")
+        repo = os.environ.get("HF_MODEL_REPO", "Angione-Lab/Shape2Force")
         files = list_repo_files(repo)
         pth_files = [f for f in files if f.startswith("ckp/") and f.endswith(".pth")]
         # For spheroid: only download ckp_spheroid_FN.pth (not ckp_spheroid_GN.pth or others)