have the patience of a saint..

gradio_app.py

@@ -3,6 +3,7 @@ import torch
 import os
 import time
 import datetime
+import shutil
 from pathlib import Path
 from moviepy.editor import VideoFileClip
 import gradio as gr
@@ -25,6 +26,20 @@ subfolders = [
 for subfolder in subfolders:
     (WEIGHTS_ROOT / subfolder).mkdir(parents=True, exist_ok=True)
 
+# Make sure legacy code that references ./weights still works by linking it to the persistent cache root.
+workspace_weights = Path("./weights")
+try:
+    if WEIGHTS_ROOT.resolve() != workspace_weights.resolve():
+        if workspace_weights.exists():
+            if workspace_weights.is_symlink() or workspace_weights.is_file():
+                workspace_weights.unlink()
+            else:
+                shutil.rmtree(workspace_weights)
+        workspace_weights.symlink_to(WEIGHTS_ROOT, target_is_directory=True)
+except FileNotFoundError:
+    # resolve() can raise if the symlink target is missing; ignore until directories exist
+    pass
+
 snapshot_download(repo_id="lixiaowen/diffuEraser", local_dir=str(WEIGHTS_ROOT / "diffuEraser"))
 snapshot_download(repo_id="stable-diffusion-v1-5/stable-diffusion-v1-5", local_dir=str(WEIGHTS_ROOT / "stable-diffusion-v1-5"))
 snapshot_download(repo_id="wangfuyun/PCM_Weights", local_dir=str(WEIGHTS_ROOT / "PCM_Weights"))