Use persistent /workspace for checkpoint cache if available

app.py

@@ -49,10 +49,14 @@ def get_checkpoint_path(checkpoint_file, default_local_path, repo_id=None, filen
     if repo_id and filename and HF_HUB_AVAILABLE:
         try:
             print(f"📥 Downloading checkpoint from {repo_id}...")
+            # Use /workspace if available (persistent), otherwise /tmp
+            cache_dir = "/workspace/checkpoints" if os.path.exists("/workspace") else "/tmp/checkpoints"
+            os.makedirs(cache_dir, exist_ok=True)
+            
             checkpoint_path = hf_hub_download(
                 repo_id=repo_id,
                 filename=filename,
-                cache_dir="/tmp/checkpoints",
+                cache_dir=cache_dir,
                 local_files_only=False
             )
             print(f"✅ Checkpoint downloaded to: {checkpoint_path}")