Update app.py

app.py

@@ -8,6 +8,7 @@ import numpy as np
 import spaces
 from omegaconf import OmegaConf
 from huggingface_hub import snapshot_download
+from PIL import Image
 
 REPO_DIR = snapshot_download(repo_id="Hang2991/PICS")
 os.chdir(REPO_DIR)
@@ -87,6 +88,9 @@ def process_composition(item, obj_thr):
     item.update({'hint': collage_final.copy()})
     return item
 
+def load_example_pil(path):
+    # 因为你已经执行了 os.chdir(REPO_DIR)，所以直接读取相对路径
+    return Image.open(path).convert("RGB")
 
 @spaces.GPU(duration=120)
 def pics_pairwise_inference(background, img_a, mask_a, img_b, mask_b):
@@ -189,18 +193,32 @@ with gr.Blocks(title="PICS: Pairwise Spatial Compositing with Spatial Interactio
 
     # --- 核心修改：把 Examples 放在这里 ---
     gr.Markdown("### 💡 Quick Examples")
+    # gr.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"
+    #         ]
+    #     ],
+    #     inputs=[bg_input, obj_a_img, obj_a_mask, obj_b_img, obj_b_mask],
+    #     outputs=output_img,   # 现在它认识这个变量了
+    #     fn=pics_pairwise_inference, 
+    #     cache_examples=True, 
+    # )
     gr.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"
+                load_example_pil("sample/bread_basket/image.jpg"), 
+                load_example_pil("sample/bread_basket/object_0.png"), 
+                load_example_pil("sample/bread_basket/object_0_mask.png"), 
+                load_example_pil("sample/bread_basket/object_1.png"), 
+                load_example_pil("sample/bread_basket/object_1_mask.png")
             ]
         ],
         inputs=[bg_input, obj_a_img, obj_a_mask, obj_b_img, obj_b_mask],
-        outputs=output_img,   # 现在它认识这个变量了
-        fn=pics_pairwise_inference, 
-        cache_examples=True, 
+        # 注意：这里去掉了 outputs 和 fn，先只做“点一下填入图片”的功能
+        cache_examples=False, 
     )
 
     # 按钮绑定