Update app.py

app.py

@@ -36,6 +36,12 @@ def load_models():
         
         print(f"✅ Models downloaded to: {repo_path}")
         
+        # Create NSFW placeholder
+        nsfw_path = "/app/CatVTON/resource/img/NSFW.jpg"
+        os.makedirs(os.path.dirname(nsfw_path), exist_ok=True)
+        if not os.path.exists(nsfw_path):
+            Image.new('RGB', (512, 512), color='black').save(nsfw_path)
+        
         print("Initializing pipeline...")
         pipeline = CatVTONPipeline(
             base_ckpt="booksforcharlie/stable-diffusion-inpainting",
@@ -48,12 +54,6 @@ def load_models():
         print("✅ Pipeline loaded!")
         
         print("Initializing automasker...")
-        
-        # Force CPU for DensePose config
-        import sys
-        sys.path.insert(0, os.path.join(repo_path, "DensePose"))
-        from detectron2.config import get_cfg
-        
         automasker = AutoMasker(
             densepose_ckpt=os.path.join(repo_path, "DensePose"),
             schp_ckpt=os.path.join(repo_path, "SCHP"),