Fix cache cleanup: wrap removals in try/except, also clear .locks placeholder

image_generator.py

@@ -41,14 +41,18 @@ if IS_HF_SPACE and os.path.isdir("/data"):
     os.environ.setdefault("HF_HOME", _cache_dir)
     print(f"Persistent cache active → {_cache_dir}")
 
-    # Clean up any stale placeholder files left by a previous failed download
-    # (huggingface_hub creates placeholder files that block re-download)
+    # Clean up stale placeholder files left by previous failed downloads.
+    # huggingface_hub creates files where it expects directories — remove them.
+    import shutil
     _model_cache = os.path.join(_cache_dir, "models--black-forest-labs--FLUX.2-klein-4B")
-    for _placeholder in [_model_cache, f"{_model_cache}/blobs", f"{_model_cache}/refs",
-                         f"{_model_cache}/snapshots"]:
-        if os.path.isfile(_placeholder):
-            os.remove(_placeholder)
-            print(f"Removed stale placeholder: {_placeholder}")
+    _locks_path  = os.path.join(_cache_dir, ".locks", "models--black-forest-labs--FLUX.2-klein-4B")
+    for _stale in [_model_cache, _locks_path]:
+        try:
+            if os.path.isfile(_stale):
+                os.remove(_stale)
+                print(f"Removed stale placeholder file: {_stale}")
+        except Exception as _e:
+            print(f"⚠️  Could not remove {_stale}: {_e}")
 
     if os.path.isdir(_model_cache):
         print(f"✅ FLUX.2-klein-4B already cached — skipping download")