Update app.py

app.py

@@ -15,6 +15,14 @@ sys.path.insert(0, REPO_DIR)
 sys.path.insert(0, os.path.join(REPO_DIR, "dinov2"))
 sys.path.insert(0, os.path.join(REPO_DIR, "sample"))
 
+
+import shutil
+# 1. 强制把 sample 文件夹拷贝到 Gradio 运行的根目录 (/home/user/app)
+# 不管你 chdir 去了哪，这一步保证了 /home/user/app/sample 真实存在
+if not os.path.exists("/home/user/app/sample"):
+    # REPO_DIR 是你下载的那个缓存路径
+    shutil.copytree(os.path.join(REPO_DIR, "sample"), "/home/user/app/sample")
+
 from cldm.model import create_model, load_state_dict
 from cldm.ddim_hacked import DDIMSampler
 from datasets.data_utils import *
@@ -23,6 +31,7 @@ model = create_model(config.config_file).cpu()
 model.load_state_dict(load_state_dict(config.pretrained_model, location='cpu'))
 model.eval()
 
+
 def get_input(batch, k):
     x = batch[k]
     if len(x.shape) == 3:
@@ -184,11 +193,13 @@ with gr.Blocks(title="PICS: Pairwise Spatial Compositing with Spatial Interactio
     # --- 核心修改：把 Examples 放在这里 ---
     gr.Markdown("### 💡 Quick Examples")
     gr.Examples(
-        examples=[
+    examples=[
             [
-                "sample/bread_basket/image.jpg", 
-                "sample/bread_basket/object_0.png", "sample/bread_basket/object_0_mask.png", 
-                "sample/bread_basket/object_1.png", "sample/bread_basket/object_1_mask.png"
+                "/home/user/app/sample/bread_basket/image.jpg", 
+                "/home/user/app/sample/bread_basket/object_0.png", 
+                "/home/user/app/sample/bread_basket/object_0_mask.png", 
+                "/home/user/app/sample/bread_basket/object_1.png", 
+                "/home/user/app/sample/bread_basket/object_1_mask.png"
             ]
         ],
         inputs=[bg_input, obj_a_img, obj_a_mask, obj_b_img, obj_b_mask],