Update app.py

app.py

@@ -26,13 +26,20 @@ REPO_ID = f"{USER_NAME}/{PRIVATE_SPACE_NAME}"
 REPO_TYPE = "space"
 
 # sys.path.append(repo_dir)
-
+cache_dir = "private_space_cache"
+if os.path.exists(cache_dir):
+    for root, dirs, files in os.walk(cache_dir):
+        for file in files:
+            if file.endswith(".incomplete"):
+                st.warning("🚨 Incomplete cache detected. Clearing cache for stability...")
+                shutil.rmtree(cache_dir)
+                break  # Stop looping once cache is cleared
 # Download the entire space, including the fine-tuned model folder
 repo_dir = snapshot_download(
     repo_id=REPO_ID,
     repo_type=REPO_TYPE,
     token=HF_TOKEN,
-    cache_dir="private_space_cache",
+    cache_dir=cache_dir, #"private_space_cache"
     force_download=True  # Forces redownload
 )