changed dockerfile

Dockerfile

@@ -22,7 +22,7 @@ RUN pip install --no-cache-dir \
     huggingface_hub
 
 # Copy app code (chown for HF Spaces permissions)
-COPY --chown=user:user app.py predictor.py ./
+COPY --chown=user:user app.py predictor.py download_ckp.py ./
 COPY --chown=user:user models/ models/
 COPY --chown=user:user utils/ utils/
 COPY --chown=user:user config/ config/
@@ -33,25 +33,7 @@ RUN mkdir -p ckp && chown user:user ckp
 # Set HF_MODEL_REPO env to your model repo, e.g. kaveh/Shape2Force
 ARG HF_MODEL_REPO=kaveh/Shape2Force
 ENV HF_MODEL_REPO=${HF_MODEL_REPO}
-
-RUN python -c "
-import os
-from pathlib import Path
-ckp = Path('/app/ckp')
-if not list(ckp.glob('*.pth')):
-    try:
-        from huggingface_hub import hf_hub_download, list_repo_files
-        repo = os.environ.get('HF_MODEL_REPO', 'kaveh/Shape2Force')
-        files = list_repo_files(repo)
-        pth_files = [f for f in files if f.startswith('ckp/') and f.endswith('.pth')]
-        for f in pth_files:
-            hf_hub_download(repo_id=repo, filename=f, local_dir='/app')
-        print('Downloaded checkpoints from', repo)
-    except Exception as e:
-        print('Could not download checkpoints:', e)
-else:
-    print('Checkpoints already present')
-"
+RUN python download_ckp.py
 
 # Ensure ckp contents are readable by user
 RUN chown -R user:user ckp

download_ckp.py

@@ -0,0 +1,19 @@
+"""Download checkpoints from Hugging Face if ckp is empty (for Space deployment)."""
+import os
+from pathlib import Path
+
+ckp = Path("/app/ckp")
+if not list(ckp.glob("*.pth")):
+    try:
+        from huggingface_hub import hf_hub_download, list_repo_files
+
+        repo = os.environ.get("HF_MODEL_REPO", "kaveh/Shape2Force")
+        files = list_repo_files(repo)
+        pth_files = [f for f in files if f.startswith("ckp/") and f.endswith(".pth")]
+        for f in pth_files:
+            hf_hub_download(repo_id=repo, filename=f, local_dir="/app")
+        print("Downloaded checkpoints from", repo)
+    except Exception as e:
+        print("Could not download checkpoints:", e)
+else:
+    print("Checkpoints already present")