Update app.py

app.py

@@ -22,36 +22,48 @@ def load_models():
     """Load models once at startup"""
     global pipeline, automasker
     
-    if pipeline is not None:
+    if pipeline is not None and automasker is not None:
         return
     
     print("🔄 Downloading/Loading CatVTON models (first time may take 5-10 mins)...")
     
-    # Download and cache models
-    repo_path = snapshot_download(
-        repo_id="zhengchong/CatVTON",
-        cache_dir="/tmp/models"  # For CPU basic tier
-    )
-    
-    print("✅ Models downloaded! Initializing pipeline...")
-    
-    # Initialize pipeline for CPU
-    pipeline = CatVTONPipeline(
-        base_ckpt="booksforcharlie/stable-diffusion-inpainting",
-        attn_ckpt=repo_path,
-        attn_ckpt_version="mix",
-        weight_dtype=torch.float32,  # CPU uses fp32
-        use_tf32=False,  # CPU doesn't support TF32
-        device='cpu'
-    )
-    
-    automasker = AutoMasker(
-        densepose_ckpt=os.path.join(repo_path, "DensePose"),
-        schp_ckpt=os.path.join(repo_path, "SCHP"),
-        device='cpu'
-    )
-    
-    print("✅ Models loaded successfully!")
+    try:
+        # Download and cache models
+        repo_path = snapshot_download(
+            repo_id="zhengchong/CatVTON",
+            cache_dir="/tmp/models"
+        )
+        
+        print(f"✅ Models downloaded to: {repo_path}")
+        print(f"DensePose path exists: {os.path.exists(os.path.join(repo_path, 'DensePose'))}")
+        print(f"SCHP path exists: {os.path.exists(os.path.join(repo_path, 'SCHP'))}")
+        
+        print("Initializing pipeline...")
+        pipeline = CatVTONPipeline(
+            base_ckpt="booksforcharlie/stable-diffusion-inpainting",
+            attn_ckpt=repo_path,
+            attn_ckpt_version="mix",
+            weight_dtype=torch.float32,
+            use_tf32=False,
+            device='cpu'
+        )
+        print("✅ Pipeline loaded!")
+        
+        print("Initializing automasker...")
+        automasker = AutoMasker(
+            densepose_ckpt=os.path.join(repo_path, "DensePose"),
+            schp_ckpt=os.path.join(repo_path, "SCHP"),
+            device='cpu'
+        )
+        print("✅ Automasker loaded!")
+        
+        print(f"Final check - Pipeline: {pipeline is not None}, Automasker: {automasker is not None}")
+        
+    except Exception as e:
+        print(f"❌ Error loading models: {e}")
+        import traceback
+        traceback.print_exc()
+        raise
 
 def generate_tryon(person_img, cloth_img, progress=gr.Progress()):
     """Generate virtual try-on"""